asigalov61
commited on
Commit
•
95f930b
1
Parent(s):
2dceebe
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ def GenerateSong(input_melody_seed_number):
|
|
81 |
x = (torch.tensor(seed_melody, dtype=torch.long, device='cuda')[None, ...])
|
82 |
|
83 |
with ctx:
|
84 |
-
with torch.inference_mode()
|
85 |
out = model.generate(x,
|
86 |
1024,
|
87 |
filter_logits_fn=top_k,
|
|
|
81 |
x = (torch.tensor(seed_melody, dtype=torch.long, device='cuda')[None, ...])
|
82 |
|
83 |
with ctx:
|
84 |
+
with torch.inference_mode():
|
85 |
out = model.generate(x,
|
86 |
1024,
|
87 |
filter_logits_fn=top_k,
|