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
|
||||
|
||||
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"
|
||||
dir_model = tmp_model_path
|
||||
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)
|
||||
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"
|
||||
if tmp_model_path.is_dir():
|
||||
print(f"{tmp_model_path} exists as a weighted model.")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue