fffiloni commited on
Commit
c432d05
·
1 Parent(s): c7d8eef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16)
7
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
8
  model = model.to(device)
9
 
 
 
 
10
  # Use bettertransform for flash attention
11
  model = BetterTransformer.transform(model, keep_original_model=False)
12
 
 
7
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
8
  model = model.to(device)
9
 
10
+ from transformers import AutoProcessor
11
+ processor = AutoProcessor.from_pretrained("suno/bark-small")
12
+
13
  # Use bettertransform for flash attention
14
  model = BetterTransformer.transform(model, keep_original_model=False)
15