Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -66,7 +66,7 @@ Lst=['input_ids',
|
|
| 66 |
'labels_attention_mask',
|
| 67 |
'mel_scaled_input_features']
|
| 68 |
def covert_cuda_batch(d):
|
| 69 |
-
|
| 70 |
for key in Lst:
|
| 71 |
d[key]=d[key].cuda(non_blocking=True)
|
| 72 |
# for key in d['text_encoder_output']:
|
|
|
|
| 66 |
'labels_attention_mask',
|
| 67 |
'mel_scaled_input_features']
|
| 68 |
def covert_cuda_batch(d):
|
| 69 |
+
return d
|
| 70 |
for key in Lst:
|
| 71 |
d[key]=d[key].cuda(non_blocking=True)
|
| 72 |
# for key in d['text_encoder_output']:
|