Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
fa88557
1
Parent(s):
34fa983
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
|