|
@@ -117,7 +117,7 @@ def load_model(model_name):
|
|
|
print(f"Could not find {pt_model}, exiting...")
|
|
print(f"Could not find {pt_model}, exiting...")
|
|
|
exit()
|
|
exit()
|
|
|
|
|
|
|
|
- model = load_quant(path_to_model, Path(f"{pt_path}"), bits)
|
|
|
|
|
|
|
+ model = load_quant(path_to_model, pt_path, bits)
|
|
|
|
|
|
|
|
# Multi-GPU setup
|
|
# Multi-GPU setup
|
|
|
if shared.args.gpu_memory:
|
|
if shared.args.gpu_memory:
|