OpenGVLab
/

InternVL2_5-1B

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Metrics Training metrics Community

Update README.md

#3

by cyrilvallez HF staff - opened 19 days ago

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -237,6 +237,7 @@ def split_model(model_name):
     device_map['mlp1'] = 0
     device_map['language_model.model.tok_embeddings'] = 0
     device_map['language_model.model.embed_tokens'] = 0
     device_map['language_model.output'] = 0
     device_map['language_model.model.norm'] = 0
     device_map['language_model.lm_head'] = 0

     device_map['mlp1'] = 0
     device_map['language_model.model.tok_embeddings'] = 0
     device_map['language_model.model.embed_tokens'] = 0
+    device_map['language_model.model.rotary_emb'] = 0
     device_map['language_model.output'] = 0
     device_map['language_model.model.norm'] = 0
     device_map['language_model.lm_head'] = 0