Spaces:

asigalov61
/

Text-to-Music-Transformer

Running

asigalov61 commited on Jul 2, 2024

Commit

bcefdd2

verified ·

1 Parent(s): b37dd07

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def Text_to_Music(input_title, input_num_tokens, input_prompt_type):
     SEQ_LEN = 4096 # Models seq len
     PAD_IDX = 2571 # Models pad index
-    DEVICE = 'cuda' # 'cuda'
     # instantiate the model
@@ -171,7 +171,7 @@ def Text_to_Music(input_title, input_num_tokens, input_prompt_type):
     inp = [outy] * number_of_batches_to_generate
-    inp = torch.LongTensor(inp).cuda()
     with ctx:
       out = model.generate(inp,

     SEQ_LEN = 4096 # Models seq len
     PAD_IDX = 2571 # Models pad index
+    DEVICE = 'cpu' # 'cuda'
     # instantiate the model
     inp = [outy] * number_of_batches_to_generate
+    inp = torch.LongTensor(inp).to(DEVICE)
     with ctx:
       out = model.generate(inp,