Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ var_2="nota-ai/bk-sdm-small"
|
|
| 58 |
|
| 59 |
|
| 60 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 61 |
-
|
| 62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
| 63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
| 64 |
pipe = pipe.to(device)
|
|
|
|
| 58 |
|
| 59 |
|
| 60 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 61 |
+
var_1, torch_dtype=torch_dtype, use_safetensors=True)
|
| 62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
| 63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
| 64 |
pipe = pipe.to(device)
|