YuE

Runtime error

fffiloni commited on 30 days ago

Commit

d978354

verified ·

1 Parent(s): 14cf1ab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -136,7 +136,7 @@ def infer(genre_txt_content, lyrics_txt_content):
         "--output_dir", f"{output_dir}",
         "--cuda_idx", "0",
         "--max_new_tokens", "3000",
-        "--disable_offload_model"
     ]
     # Set up environment variables for CUDA with optimized settings

         "--output_dir", f"{output_dir}",
         "--cuda_idx", "0",
         "--max_new_tokens", "3000",
+        #"--disable_offload_model"
     ]
     # Set up environment variables for CUDA with optimized settings