Commit 25f2f756 authored by zhuwenwen's avatar zhuwenwen
Browse files

Merge branch 'v0.18.0-ori' of http://10.16.6.30/dcutoolkit/deeplearing/vllm into v0.18.0-ori

parents 6f1beee6 58beb740
......@@ -2359,7 +2359,7 @@ class NemotronH_Nano_VL_V2(
with torch.no_grad():
default_weight_loader(param, w)
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
vision_weights.append((hf_key, w))
elif is_sound_weights(name):
......@@ -2419,4 +2419,4 @@ class NemotronH_Nano_VL_V2(
@classmethod
def get_mamba_state_copy_func(cls):
return NemotronHForCausalLM.get_mamba_state_copy_func()
\ No newline at end of file
return NemotronHForCausalLM.get_mamba_state_copy_func()
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