Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ var_2="nota-ai/bk-sdm-small"
|
|
57 |
|
58 |
|
59 |
pipe = DiffusionPipeline.from_pretrained(
|
60 |
-
|
61 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
62 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
63 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
|
|
57 |
|
58 |
|
59 |
pipe = DiffusionPipeline.from_pretrained(
|
60 |
+
model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
|
61 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
62 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
63 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|