AshwinSankar commited on
Commit
51aea63
1 Parent(s): 5150d64

remove low_cpu_mem_usage

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, low_cpu_mem_usage=True
25
  ).to(device)
26
  jenny_model = ParlerTTSForConditionalGeneration.from_pretrained(
27
- jenny_repo_id, attn_implementation="eager", torch_dtype=torch_dtype, low_cpu_mem_usage=True
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)