Flux9665 commited on
Commit
5ad4ca1
·
1 Parent(s): 9c26912

try to figure out how ZeroGPU works

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def read(prompt,
136
  emb1,
137
  emb2
138
  ):
139
- with torch.inference_mode():
140
  sr, wav, fig = controllable_ui.read(prompt,
141
  language.split(" ")[-1].split("(")[1].split(")")[0],
142
  language.split(" ")[-1].split("(")[1].split(")")[0],
 
136
  emb1,
137
  emb2
138
  ):
139
+ with torch.no_grad():
140
  sr, wav, fig = controllable_ui.read(prompt,
141
  language.split(" ")[-1].split("(")[1].split(")")[0],
142
  language.split(" ")[-1].split("(")[1].split(")")[0],