Bartusito commited on
Commit
906d41e
1 Parent(s): a6c765b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -19,8 +19,7 @@ def modelo2(text):
19
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
20
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
21
  pipe = pipe.to("cpu")
22
-
23
- text = torch.tensor(text, dtype=torch.float32) # Conversión a float32
24
  image = pipe(text).images[0]
25
 
26
  return image
 
19
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
20
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
21
  pipe = pipe.to("cpu")
22
+ # Conversión a float32
 
23
  image = pipe(text).images[0]
24
 
25
  return image