Import fix

This commit is contained in:
somebody
2023-07-24 13:09:15 -05:00
parent a6aafb2525
commit 4a6cccb002

View File

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