Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|