rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Import fix

Browse Source
This commit is contained in:
somebody
2023-07-24 13:09:15 -05:00
parent a6aafb2525
commit 4a6cccb002
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modeling/inference_models/gptq_hf_torch/class.py
View File

@@ -90,7 +90,6 @@ class model_backend(HFTorchInferenceModel):
def _load(self, save_model: bool, initial_load: bool) -> None: def _load(self, save_model: bool, initial_load: bool) -> None:
try: try:
import hf_bleeding_edge
from hf_bleeding_edge import AutoModelForCausalLM from hf_bleeding_edge import AutoModelForCausalLM
except ImportError: except ImportError:
from transformers import AutoModelForCausalLM from transformers import AutoModelForCausalLM