Old-Fisherman commited on
Commit
adfffe7
·
verified ·
1 Parent(s): b922c99

Update convert_mod.py

Browse files
Files changed (1) hide show
  1. convert_mod.py +3 -3
convert_mod.py CHANGED
@@ -77,10 +77,10 @@ class ModelSD1(ModelTemplate):
77
  class ModelClipG(ModelTemplate):
78
  arch = "clip_g"
79
  keys_detect = [
80
- ("text_model.encoder.layers.22.ln_final.bias",), # Example key, adjust as needed
81
  (
82
- "text_model.encoder.layers.0.attn.in_proj_weight", # Example key, adjust as needed
83
- "text_model.encoder.layers.1.attn.in_proj_weight", # Example key, adjust as needed
84
  ),
85
  ]
86
  keys_banned = [] # Add any banned keys if necessary
 
77
  class ModelClipG(ModelTemplate):
78
  arch = "clip_g"
79
  keys_detect = [
80
+ ("text_model.final_layer_norm.bias",), # Example key, adjust as needed
81
  (
82
+ "text_model.encoder.layers.0.self_attn.k_proj.weight", # Example key, adjust as needed
83
+ "text_model.encoder.layers.1.self_attn.q_proj.bias", # Example key, adjust as needed
84
  ),
85
  ]
86
  keys_banned = [] # Add any banned keys if necessary