Minor change

This commit is contained in:
oobabooga 2023-05-17 11:33:43 -03:00
parent 21ecc3701e
commit abd361b3a0

View File

@ -71,7 +71,7 @@ def load_model(model_name):
t0 = time.time() t0 = time.time()
shared.model_type = find_model_type(model_name) shared.model_type = find_model_type(model_name)
if shared.args.wbits > 0: if shared.args.wbits > 0 or shared.args.autogptq:
if shared.args.autogptq: if shared.args.autogptq:
load_func = AutoGPTQ_loader load_func = AutoGPTQ_loader
else: else: