Spaces:
Running
on
Zero
Running
on
Zero
AshwinSankar
commited on
Commit
•
51aea63
1
Parent(s):
5150d64
remove low_cpu_mem_usage
Browse files
app.py
CHANGED
@@ -21,10 +21,10 @@ repo_id = "ai4bharat/indic-parler-tts-pretrained"
|
|
21 |
jenny_repo_id = "ai4bharat/indic-parler-tts"
|
22 |
|
23 |
model = ParlerTTSForConditionalGeneration.from_pretrained(
|
24 |
-
repo_id, attn_implementation="eager", torch_dtype=torch_dtype,
|
25 |
).to(device)
|
26 |
jenny_model = ParlerTTSForConditionalGeneration.from_pretrained(
|
27 |
-
jenny_repo_id, attn_implementation="eager", torch_dtype=torch_dtype,
|
28 |
).to(device)
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
|
|
21 |
jenny_repo_id = "ai4bharat/indic-parler-tts"
|
22 |
|
23 |
model = ParlerTTSForConditionalGeneration.from_pretrained(
|
24 |
+
repo_id, attn_implementation="eager", torch_dtype=torch_dtype,
|
25 |
).to(device)
|
26 |
jenny_model = ParlerTTSForConditionalGeneration.from_pretrained(
|
27 |
+
jenny_repo_id, attn_implementation="eager", torch_dtype=torch_dtype,
|
28 |
).to(device)
|
29 |
|
30 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|