Spaces:
Running
Running
Sean Powell commited on
Commit ·
a30feef
1
Parent(s): 21ca519
Try disabling VAE slicing to see if that removes the CUDA incompatible half type.
Browse files- utils/pipes.py +2 -0
utils/pipes.py
CHANGED
|
@@ -9,6 +9,7 @@ def create_stable_diffusion_xl_pipeline(device):
|
|
| 9 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 10 |
)
|
| 11 |
pipe = pipe.to(device)
|
|
|
|
| 12 |
pipe.vae.disable_tiling()
|
| 13 |
tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder, pipe.unet])
|
| 14 |
return pipe
|
|
@@ -19,6 +20,7 @@ def create_stable_diffusion_xl_img2img_pipe(device):
|
|
| 19 |
"stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 20 |
)
|
| 21 |
pipe = pipe.to(device)
|
|
|
|
| 22 |
pipe.vae.disable_tiling()
|
| 23 |
tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder_2, pipe.unet])
|
| 24 |
return pipe
|
|
|
|
| 9 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 10 |
)
|
| 11 |
pipe = pipe.to(device)
|
| 12 |
+
pipe.vae.disable_vae_slicing() # Experimental
|
| 13 |
pipe.vae.disable_tiling()
|
| 14 |
tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder, pipe.unet])
|
| 15 |
return pipe
|
|
|
|
| 20 |
"stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 21 |
)
|
| 22 |
pipe = pipe.to(device)
|
| 23 |
+
pipe.vae.disable_vae_slicing() # Experimental
|
| 24 |
pipe.vae.disable_tiling()
|
| 25 |
tiling.enable_circular_tiling([pipe.vae, pipe.text_encoder_2, pipe.unet])
|
| 26 |
return pipe
|