Skip to content

GitLab

Commit 58beb740 authored by zhuwenwen's avatar zhuwenwen
Browse files

Update nano_nemotron_vl.py

parent eefa41c1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/model_executor/models/nano_nemotron_vl.py
View file @ 58beb740
...@@ -2359,7 +2359,7 @@ class NemotronH_Nano_VL_V2( ...@@ -2359,7 +2359,7 @@ class NemotronH_Nano_VL_V2(
with torch.no_grad(): with torch.no_grad():
default_weight_loader(param, w) default_weight_loader(param, w)
elif is_vision_weights(name): elif is_vision_weights(name):
# Convert: vision_model.radio_model.* 鈫?radio_model.* # Convert: vision_model.radio_model.* → radio_model.*
hf_key = name[len("vision_model.") :] # Remove "vision_model." prefix hf_key = name[len("vision_model.") :] # Remove "vision_model." prefix
vision_weights.append((hf_key, w)) vision_weights.append((hf_key, w))
elif is_sound_weights(name): elif is_sound_weights(name):
...@@ -2419,4 +2419,4 @@ class NemotronH_Nano_VL_V2( ...@@ -2419,4 +2419,4 @@ class NemotronH_Nano_VL_V2(
@classmethod @classmethod
def get_mamba_state_copy_func(cls): def get_mamba_state_copy_func(cls):
return NemotronHForCausalLM.get_mamba_state_copy_func() return NemotronHForCausalLM.get_mamba_state_copy_func()
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment