mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

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

Browse source
This commit is contained in:
oobabooga 2023-03-13 00:08:01 -03:00
parent b9e0712b92
commit 77294b27dd
3 changed files with 7 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/quantized_LLaMA.py
View file

@ -1,4 +1,3 @@
import os
import sys
from pathlib import Path
@ -7,7 +6,7 @@ import torch
import modules.shared as shared
sys.path.insert(0, os.path.abspath(Path("repositories/GPTQ-for-LLaMa")))
sys.path.insert(0, str(Path("repositories/GPTQ-for-LLaMa")))
from llama import load_quant
@ -41,7 +40,7 @@ def load_quantized_LLaMA(model_name):
print(f"Could not find {pt_model}, exiting...")
exit()
model = load_quant(path_to_model, os.path.abspath(pt_path), bits)
model = load_quant(path_to_model, str(pt_path), bits)
# Multi-GPU setup
if shared.args.gpu_memory: