Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ tiling, trimmed, unfocused, unattractive, unnatural pose, unreal engine, unsight
|
|
60 |
|
61 |
|
62 |
pipe = StableDiffusionPipeline.from_pretrained(
|
63 |
-
"
|
64 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|
|
|
60 |
|
61 |
|
62 |
pipe = StableDiffusionPipeline.from_pretrained(
|
63 |
+
"nota-ai/bk-sdm-small", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|