update: change folder architecture
This commit is contained in:
parent
b00e2d90f9
commit
440cc2f46f
5 changed files with 4 additions and 2 deletions
|
@ -1040,7 +1040,8 @@ args = parse_args()
|
||||||
dir_model = args.model
|
dir_model = args.model
|
||||||
|
|
||||||
if args.awq_path:
|
if args.awq_path:
|
||||||
from awqpy.apply_awq import add_scale_weights
|
sys.path.insert(1, str(Path(__file__).parent / 'awq-py'))
|
||||||
|
from awq.apply_awq import add_scale_weights
|
||||||
tmp_model_path = args.model / "weighted_model"
|
tmp_model_path = args.model / "weighted_model"
|
||||||
dir_model = tmp_model_path
|
dir_model = tmp_model_path
|
||||||
if tmp_model_path.is_dir():
|
if tmp_model_path.is_dir():
|
||||||
|
|
|
@ -1200,7 +1200,8 @@ def main(args_in: list[str] | None = None) -> None:
|
||||||
|
|
||||||
args = parser.parse_args(args_in)
|
args = parser.parse_args(args_in)
|
||||||
if args.awq_path:
|
if args.awq_path:
|
||||||
from awqpy.apply_awq import add_scale_weights
|
sys.path.insert(1, str(Path(__file__).parent / 'awq-py'))
|
||||||
|
from awq.apply_awq import add_scale_weights
|
||||||
tmp_model_path = args.model / "weighted_model"
|
tmp_model_path = args.model / "weighted_model"
|
||||||
if tmp_model_path.is_dir():
|
if tmp_model_path.is_dir():
|
||||||
print(f"{tmp_model_path} exists as a weighted model.")
|
print(f"{tmp_model_path} exists as a weighted model.")
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue