Update model.py

SaladTechnologies · Dec 18, 2023 · d0a0393 · d0a0393
1 parent 3fa1e56
commit d0a0393
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llama-7b/model/model.py b/src/llama-7b/model/model.py
@@ -13,7 +13,7 @@ def __init__(self, **kwargs) -> None:
         self._tokenizer = None
 
     def load(self):
-        self._tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
+        self._tokenizer = LlamaTokenizer.from_pretrained("huggyllama/llama-7b")
         self._model = LlamaForCausalLM.from_pretrained(
             str(self._data_dir),
             torch_dtype=torch.float16,