Browse Source

Don't initialize clip weights to default values.

pull/770/head
comfyanonymous 1 year ago
parent
commit
0c7cad404c
  1. 3
      comfy/sd1_clip.py

3
comfy/sd1_clip.py

@ -1,6 +1,6 @@
import os
from transformers import CLIPTokenizer, CLIPTextModel, CLIPTextConfig
from transformers import CLIPTokenizer, CLIPTextModel, CLIPTextConfig, modeling_utils
import torch
import traceback
import zipfile
@ -38,6 +38,7 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder):
if textmodel_json_config is None:
textmodel_json_config = os.path.join(os.path.dirname(os.path.realpath(__file__)), "sd1_clip_config.json")
config = CLIPTextConfig.from_json_file(textmodel_json_config)
with modeling_utils.no_init_weights():
self.transformer = CLIPTextModel(config)
self.device = device

Loading…
Cancel
Save