File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
tests/unittest/_torch/modeling Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 44
55import torch
66from parameterized import parameterized
7+ from utils .llm_data import llm_models_root
78
89import tensorrt_llm
910from tensorrt_llm ._torch .attention_backend .utils import get_attention_backend
1617from tensorrt_llm .mapping import Mapping
1718
1819VILA_1_5_3B_CONFIG = {
19- "_name_or_path" : "Efficient-Large-Model /VILA1.5-3b" ,
20+ "_name_or_path" : f" { llm_models_root () } /vila /VILA1.5-3b" ,
2021 "architectures" : ["LlavaLlamaModel" ],
2122 "drop_path_rate" : 0.0 ,
2223 "hidden_size" : 2560 ,
263264
264265NVILA_8B_CONFIG = {
265266 "_attn_implementation_autoset" : True ,
266- "_name_or_path" : "Efficient-Large-Model /NVILA-8B" ,
267+ "_name_or_path" : f" { llm_models_root () } /vila /NVILA-8B" ,
267268 "architectures" : ["LlavaLlamaModel" ],
268269 "chat_template" : None ,
269270 "drop_path_rate" : 0.0 ,
You can’t perform that action at this time.
0 commit comments