Hugging Face's logo

Spaces:
wasmdashai
/
RunTasking
Running on Zero

App Files Community
wasmdashai commited on
Commit
7f59922
·
verified ·
1 Parent(s): 1f1eee1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- # return d
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']: