yeq6x commited on
Commit
6ea5553
·
verified ·
1 Parent(s): d032063

Update scripts/process_utils.py

Browse files
Files changed (1) hide show
  1. scripts/process_utils.py +1 -1
scripts/process_utils.py CHANGED
@@ -135,7 +135,7 @@ def initialize_refine_model():
135
 
136
  # Create the ControlNet pipeline
137
  refine_gen_pipe = StableDiffusionControlNetPipeline(
138
- vae=AutoencoderKL.from_single_file(vae_path, torch_dtype=torch_dtype).to(device),
139
  text_encoder=sd_pipe.text_encoder,
140
  tokenizer=sd_pipe.tokenizer,
141
  unet=sd_pipe.unet,
 
135
 
136
  # Create the ControlNet pipeline
137
  refine_gen_pipe = StableDiffusionControlNetPipeline(
138
+ vae=AutoencoderKL.from_single_file(vae_path, torch_dtype=torch_dtype, from_tf=True).to(device),
139
  text_encoder=sd_pipe.text_encoder,
140
  tokenizer=sd_pipe.tokenizer,
141
  unet=sd_pipe.unet,