Spaces:

Giang07
/

Ti6-lllama2-QLoRra

Paused

Giang07 commited on May 14

Commit

9787fba

•

1 Parent(s): 51f3cc1

Update model_utils.py

Files changed (1) hide show

model_utils.py CHANGED Viewed

@@ -39,8 +39,8 @@ def load_model():
     # else:
     #     model = AutoModelForCausalLM.from_pretrained(MODEL_DIR)
     #     tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
-    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
     return model, tokenizer

     # else:
     #     model = AutoModelForCausalLM.from_pretrained(MODEL_DIR)
     #     tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
+    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME, trust_remote_code=True)
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
     return model, tokenizer