Update app.py
Browse files
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 |
|