Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
997d449
1
Parent(s):
b6de3f2
fix cuda initialization for zerogpu
Browse files
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())
|