Update app.py
Browse files
app.py
CHANGED
@@ -240,7 +240,7 @@ def generate_music(
|
|
240 |
use_cache=True,
|
241 |
top_k=50,
|
242 |
num_beams=1,
|
243 |
-
assistant_model=assistant_model
|
244 |
)
|
245 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
246 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|
|
|
240 |
use_cache=True,
|
241 |
top_k=50,
|
242 |
num_beams=1,
|
243 |
+
# assistant_model=assistant_model
|
244 |
)
|
245 |
if output_seq[0][-1].item() != mmtokenizer.eoa:
|
246 |
tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
|