preston-cell commited on
Commit
a80995d
·
verified ·
1 Parent(s): 4f03078

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -4,8 +4,11 @@ import numpy as np
4
  from generator import load_csm_1b
5
  import torchaudio
6
 
7
- # Load CSM model
8
- generator = load_csm_1b(device="cpu")
 
 
 
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")