Commit
•
82ba5a9
1
Parent(s):
041c40c
bug fix
Browse files
app.py
CHANGED
|
@@ -30,8 +30,8 @@ from transformers import LlamaTokenizer
|
|
| 30 |
from transformers import PreTrainedModel
|
| 31 |
from transformers import PreTrainedTokenizerBase
|
| 32 |
|
| 33 |
-
transformers.AutoTokenizer = LlamaTokenizer
|
| 34 |
-
transformers.AutoModelForCausalLM = LlamaForCausalLM
|
| 35 |
|
| 36 |
def load_lora_model(
|
| 37 |
model_path: str,
|
|
|
|
| 30 |
from transformers import PreTrainedModel
|
| 31 |
from transformers import PreTrainedTokenizerBase
|
| 32 |
|
| 33 |
+
transformers.AutoTokenizer.from_pretrained = LlamaTokenizer.from_pretrained
|
| 34 |
+
transformers.AutoModelForCausalLM.from_pretrained = LlamaForCausalLM.from_pretrained
|
| 35 |
|
| 36 |
def load_lora_model(
|
| 37 |
model_path: str,
|