zhouzaida commited on
Commit
d869dc5
·
1 Parent(s): 7718375
Files changed (1) hide show
  1. modeling_kimi_vl.py +1 -0
modeling_kimi_vl.py CHANGED
@@ -2263,6 +2263,7 @@ class MoonVitPretrainedModel(PreTrainedModel):
2263
  _no_split_modules = ["PackingTransformer"]
2264
  _supports_flash_attn_2 = True
2265
  _supports_sdpa = True
 
2266
  def __init__(self, config: MoonViTConfig, *inputs, **kwargs):
2267
  super().__init__(config, *inputs, **kwargs)
2268
  config = deepcopy(config)
 
2263
  _no_split_modules = ["PackingTransformer"]
2264
  _supports_flash_attn_2 = True
2265
  _supports_sdpa = True
2266
+
2267
  def __init__(self, config: MoonViTConfig, *inputs, **kwargs):
2268
  super().__init__(config, *inputs, **kwargs)
2269
  config = deepcopy(config)