Imsachin010 commited on
Commit
29acf31
·
1 Parent(s): f5051d6

Fix indentation bug on line 42

Browse files
Files changed (1) hide show
  1. training/grpo_train.py +1 -1
training/grpo_train.py CHANGED
@@ -38,7 +38,7 @@ def _load_model_and_tokenizer(model_name: str, use_unsloth: bool = False):
38
  if use_unsloth:
39
  try:
40
  from unsloth import FastLanguageModel
41
- print("Loading with Unsloth in 4-bit + LoRA...")
42
  model, tokenizer = FastLanguageModel.from_pretrained(
43
  model_name=model_name,
44
  max_seq_length=2048,
 
38
  if use_unsloth:
39
  try:
40
  from unsloth import FastLanguageModel
41
+ print("Loading with Unsloth in 4-bit + LoRA...")
42
  model, tokenizer = FastLanguageModel.from_pretrained(
43
  model_name=model_name,
44
  max_seq_length=2048,