Update convert_mod.py
Browse files- 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.
|
81 |
(
|
82 |
-
"text_model.encoder.layers.0.
|
83 |
-
"text_model.encoder.layers.1.
|
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
|