We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 580d6df commit 74da2a7Copy full SHA for 74da2a7
fastvideo/v1/configs/models/encoders/clip.py
@@ -62,6 +62,13 @@ class CLIPVisionArchConfig(ImageEncoderArchConfig):
62
attention_dropout: float = 0.0
63
initializer_range: float = 0.02
64
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
+ ])
72
73
74
@dataclass
0 commit comments