mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-26 01:30:20 +01:00
Minor change
This commit is contained in:
parent
21ecc3701e
commit
abd361b3a0
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user