Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ pipe_edit.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_
|
|
| 34 |
pipe_edit.to("cuda")
|
| 35 |
|
| 36 |
pipe_normal = StableDiffusionXLPipeline.from_single_file(normal_file, torch_dtype=torch.float16, vae=vae)
|
| 37 |
-
pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction", sigma_schedule="exponential"
|
| 38 |
pipe_normal.to("cuda")
|
| 39 |
|
| 40 |
@spaces.GPU
|
|
|
|
| 34 |
pipe_edit.to("cuda")
|
| 35 |
|
| 36 |
pipe_normal = StableDiffusionXLPipeline.from_single_file(normal_file, torch_dtype=torch.float16, vae=vae)
|
| 37 |
+
pipe_normal.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction", sigma_schedule="exponential")
|
| 38 |
pipe_normal.to("cuda")
|
| 39 |
|
| 40 |
@spaces.GPU
|