Spaces:
Runtime error
Runtime error
Update scripts/process_utils.py
Browse files- 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,
|