Sfoglia il codice sorgente

Insert to the path instead of appending

oobabooga 2 anni fa
parent
commit
74102d5ee4
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      modules/models.py

+ 1 - 1
modules/models.py

@@ -89,7 +89,7 @@ def load_model(model_name):
 
     # 4-bit LLaMA
     elif shared.args.load_in_4bit:
-        sys.path.append(os.path.abspath(Path("repositories/GPTQ-for-LLaMa")))
+        sys.path.insert(0, os.path.abspath(Path("repositories/GPTQ-for-LLaMa")))
 
         from llama import load_quant