Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -61,8 +61,8 @@ tiling, trimmed, unfocused, unattractive, unnatural pose, unreal engine, unsight
|
|
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)
|
67 |
pipe.enable_vae_tiling()
|
68 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|
|
|
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)
|
67 |
pipe.enable_vae_tiling()
|
68 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|