Bartusito commited on
Commit
e08cc47
·
1 Parent(s): 9dc356c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,7 +21,9 @@ def modelo2(text):
21
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
22
  pipe = pipe.to("cpu")
23
 
 
24
  image = pipe(text).images[0]
 
25
  return image
26
 
27
  def execution(audio):
 
21
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
22
  pipe = pipe.to("cpu")
23
 
24
+ text = text.to(torch.float32)
25
  image = pipe(text).images[0]
26
+
27
  return image
28
 
29
  def execution(audio):