소스 검색

换用github jwjohns/unsloth-GRPO-qwen2.5 验证GRPO训练模型

zhouyang.xie 2 달 전
부모
커밋
2f61ec36fb
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      src/train_model_github_jwjohns.py

+ 1 - 0
src/train_model_github_jwjohns.py

@@ -5,6 +5,7 @@
 import re
 import torch
 import os
+import json
 from unsloth import FastLanguageModel, PatchFastRL, is_bfloat16_supported
 
 # Enable Unsloth's CLI training metrics visualization