fix bug which causes the following RuntimeError when call apply_weights…#2

hliuca merged 1 commit intoROCm:0.3.0-rocmROCm/vllm:0.3.0-rocmfrom

yaomingamd:0.3.0-rocmyaomingamd/vllm_rocm:0.3.0-rocmCopy head branch name to clipboard