electroglyph commited on
Commit
e141b33
·
verified ·
1 Parent(s): 3da3f5f

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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="/home/anon/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
 
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