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
Files changed (1) hide show
  1. 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