Spaces:

100XZX001
/

code-review-training

Sleeping

100XZX001 commited on Apr 25

Commit

374db2f

verified ·

1 Parent(s): d9cfe52

Update training.py

Files changed (1) hide show

training.py CHANGED Viewed

@@ -78,7 +78,7 @@ def map_to_env(action: AgentAction):
 def load_model():
     model, tokenizer = FastLanguageModel.from_pretrained(
     model_name="unsloth/Phi-3-mini-4k-instruct-bnb-4bit",
-    max_seq_length=768,
     load_in_4bit=True,
     )
     model = FastLanguageModel.get_peft_model(

 def load_model():
     model, tokenizer = FastLanguageModel.from_pretrained(
     model_name="unsloth/Phi-3-mini-4k-instruct-bnb-4bit",
+    max_seq_length=480,
     load_in_4bit=True,
     )
     model = FastLanguageModel.get_peft_model(