Skip to content

Commit 74da2a7

Browse files
authored
Fix CLIP config (#568)
1 parent 580d6df commit 74da2a7

File tree

1 file changed

+7
-0
lines changed
  • fastvideo/v1/configs/models/encoders

1 file changed

+7
-0
lines changed

fastvideo/v1/configs/models/encoders/clip.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,13 @@ class CLIPVisionArchConfig(ImageEncoderArchConfig):
6262
attention_dropout: float = 0.0
6363
initializer_range: float = 0.02
6464
initializer_factor: float = 1.0
65+
stacked_params_mapping: List[Tuple[str, str,
66+
str]] = field(default_factory=lambda: [
67+
# (param_name, shard_name, shard_id)
68+
("qkv_proj", "q_proj", "q"),
69+
("qkv_proj", "k_proj", "k"),
70+
("qkv_proj", "v_proj", "v"),
71+
])
6572

6673

6774
@dataclass

0 commit comments

Comments
 (0)