Update train.py
Browse files
train.py
CHANGED
@@ -961,7 +961,7 @@ class MonitorRewardCallback(TrainerCallback):
|
|
961 |
max_seq_length = 1200
|
962 |
|
963 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
964 |
-
model_name="/
|
965 |
# fast_inference = True,
|
966 |
max_seq_length=max_seq_length, # Choose any for long context!
|
967 |
load_in_4bit=False, # 4 bit quantization to reduce memory
|
|
|
961 |
max_seq_length = 1200
|
962 |
|
963 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
964 |
+
model_name="google/gemma-3-4b-it",
|
965 |
# fast_inference = True,
|
966 |
max_seq_length=max_seq_length, # Choose any for long context!
|
967 |
load_in_4bit=False, # 4 bit quantization to reduce memory
|