nodes/sylym-comfy_vid2vid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

first modification

Browse Source
This commit is contained in:
sylym
2023-03-24 22:14:49 +08:00
parent 9c2998dc2b
commit 686791ef70
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sd.py
+3
View File
@@ -104,6 +104,9 @@ def load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, e
model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to) model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to)
model.model.diffusion_model = convert_unet_checkpoint(sd, OmegaConf.create({"model": model_config})) model.model.diffusion_model = convert_unet_checkpoint(sd, OmegaConf.create({"model": model_config}))
if model_management.xformers_enabled():
model.model.diffusion_model.enable_xformers_memory_efficient_attention()
if fp16: if fp16:
model = model.half() model = model.half()