Update app.py
Browse files
app.py
CHANGED
|
@@ -106,6 +106,7 @@ inst_decoder = torch.compile(inst_decoder)
|
|
| 106 |
vocal_decoder.eval()
|
| 107 |
inst_decoder.eval()
|
| 108 |
|
|
|
|
| 109 |
|
| 110 |
def generate_music(
|
| 111 |
max_new_tokens=5,
|
|
|
|
| 106 |
vocal_decoder.eval()
|
| 107 |
inst_decoder.eval()
|
| 108 |
|
| 109 |
+
cuda_idx = 0
|
| 110 |
|
| 111 |
def generate_music(
|
| 112 |
max_new_tokens=5,
|