KaiChen1998 commited on
Commit
997d449
·
1 Parent(s): b6de3f2

fix cuda initialization for zerogpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -321,7 +321,7 @@ def http_bot(state, temperature, top_p, max_new_tokens, speaker):
321
 
322
  condition = f'gender-{gender}_emotion-{emotion}_speed-{speed}_pitch-{pitch}'
323
  style_centroid_file = condition2style_centroid_file_dict[condition]
324
- style_centroid_embedding = condition2style_centroid_embedding_dict[condition]
325
  print(condition)
326
 
327
  id = str(uuid.uuid4())
 
321
 
322
  condition = f'gender-{gender}_emotion-{emotion}_speed-{speed}_pitch-{pitch}'
323
  style_centroid_file = condition2style_centroid_file_dict[condition]
324
+ style_centroid_embedding = condition2style_centroid_embedding_dict[condition].cuda()
325
  print(condition)
326
 
327
  id = str(uuid.uuid4())