Skip to content

GitLab

"...git@developer.sourcefind.cn:dcuai/dlexamples.git" did not exist on "f0d87682bb9d7f4a2b3280eadd2c33e82687619d"
Commit 1ddf512f authored by comfyanonymous's avatar comfyanonymous
Browse files

Don't auto convert clip and vae weights to fp16 when saving checkpoint.

parent 32be3582
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
comfy/model_base.py
View file @ 1ddf512f
...@@ -207,9 +207,6 @@ class BaseModel(torch.nn.Module): ...@@ -207,9 +207,6 @@ class BaseModel(torch.nn.Module):
unet_state_dict = self.diffusion_model.state_dict() unet_state_dict = self.diffusion_model.state_dict()
unet_state_dict = self.model_config.process_unet_state_dict_for_saving(unet_state_dict) unet_state_dict = self.model_config.process_unet_state_dict_for_saving(unet_state_dict)
if self.get_dtype() == torch.float16:
extra_sds = map(lambda sd: utils.convert_sd_to(sd, torch.float16), extra_sds)
if self.model_type == ModelType.V_PREDICTION: if self.model_type == ModelType.V_PREDICTION:
unet_state_dict["v_pred"] = torch.tensor([]) unet_state_dict["v_pred"] = torch.tensor([])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment