Hugging Face's logo

OpenGVLab
/
InternVL-14B-224px

Image Feature Extraction
Transformers
PyTorch
Safetensors
internvl
feature-extraction
custom_code
Model card Files
xet
 Community
5
czczup commited on
Commit
0bffb65
·
verified ·
1 Parent(s): d707ab1

Update modeling_internvl.py

Browse files
Files changed (1) hide show
  1. modeling_internvl.py +1 -1
modeling_internvl.py CHANGED
@@ -42,7 +42,7 @@ class InternVLPreTrainedModel(PreTrainedModel):
42
  _keys_to_ignore_on_load_missing = [
43
  r'position_ids',
44
  ]
45
- _no_split_modules = ['InternAttention', 'LlamaDecoderLayer']
46
  _skip_keys_device_placement = 'past_key_values'
47
  _keep_in_fp32_modules = ['wo']
48
 
 
42
  _keys_to_ignore_on_load_missing = [
43
  r'position_ids',
44
  ]
45
+ _no_split_modules = ['InternAttention', 'LlamaDecoderLayer', 'LlamaForCausalLM']
46
  _skip_keys_device_placement = 'past_key_values'
47
  _keep_in_fp32_modules = ['wo']
48