Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,6 @@ def generate(
|
|
| 61 |
|
| 62 |
if not use_vae:
|
| 63 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
| 64 |
-
sampler_class = sampler
|
| 65 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 66 |
|
| 67 |
if use_vae:
|
|
|
|
| 61 |
|
| 62 |
if not use_vae:
|
| 63 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
|
|
|
| 64 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 65 |
|
| 66 |
if use_vae:
|