Update scripts/process_utils.py

scripts/process_utils.py

@@ -135,7 +135,7 @@ def initialize_refine_model():
 
     # Create the ControlNet pipeline
     refine_gen_pipe = StableDiffusionControlNetPipeline(
-        vae=AutoencoderKL.from_single_file(vae_path, torch_dtype=torch_dtype).to(device),
+        vae=AutoencoderKL.from_single_file(vae_path, torch_dtype=torch_dtype, from_tf=True).to(device),
         text_encoder=sd_pipe.text_encoder,
         tokenizer=sd_pipe.tokenizer,
         unet=sd_pipe.unet,