Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,11 @@ import numpy as np
|
|
4 |
from generator import load_csm_1b
|
5 |
import torchaudio
|
6 |
|
7 |
-
# Load
|
8 |
-
|
|
|
|
|
|
|
9 |
|
10 |
# Load image-to-text model
|
11 |
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
|
|
|
4 |
from generator import load_csm_1b
|
5 |
import torchaudio
|
6 |
|
7 |
+
# Load the configuration manually
|
8 |
+
config = AutoConfig.from_pretrained("sesame/csm-1b")
|
9 |
+
|
10 |
+
# Load the model with config
|
11 |
+
generator = load_csm_1b(device="cpu", config=config)
|
12 |
|
13 |
# Load image-to-text model
|
14 |
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
|