Spaces:

tonychenxyz
/

emo-knob

Runtime error

tonychenxyz commited on Oct 3, 2024

Commit

73701f9

1 Parent(s): 9bfcf61

added back device

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,7 +87,7 @@ if torch.cuda.is_available():
         seed = 1337
         output_dir = "outputs"
         _dtype = get_default_dtype()
-        # _device = 'cuda:0'
         _model_dir = snapshot_download(repo_id=model_name)
         first_stage_adapter = FlattenedInterleavedEncodec2Codebook(end_of_audio_token=1024)

         seed = 1337
         output_dir = "outputs"
         _dtype = get_default_dtype()
+        _device = 'cuda:0'
         _model_dir = snapshot_download(repo_id=model_name)
         first_stage_adapter = FlattenedInterleavedEncodec2Codebook(end_of_audio_token=1024)