asigalov61 commited on
Commit
95f930b
1 Parent(s): 2dceebe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,