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:
|