Use os.path.abspath() instead of str()

This commit is contained in:
oobabooga 2023-03-12 12:41:04 -03:00
parent 8403152257
commit fda376d9c3

View File

@ -41,7 +41,7 @@ def load_quantized_LLaMA(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, str(pt_path), bits) model = load_quant(path_to_model, os.path.abspath(pt_path), bits)
# Multi-GPU setup # Multi-GPU setup
if shared.args.gpu_memory: if shared.args.gpu_memory: