|
@@ -70,8 +70,8 @@ def load_pretrained_block(
|
|
|
hf_quantizer.preprocess_model(
|
|
|
model=tmp_block_list, device_map=device_map, keep_in_fp32_modules=False,
|
|
|
)
|
|
|
- if model_name[-1] != '.':
|
|
|
- model_name += '.'
|
|
|
+ # if model_name[-1] != '.':
|
|
|
+ # model_name += '.'
|
|
|
|
|
|
state_dict = _load_state_dict_from_repo(
|
|
|
model_name,
|