Spaces:
Sleeping
Sleeping
Commit ·
2c7592a
1
Parent(s): 8dc4077
update
Browse files
examples/tutorials/grpo/step_2_train_grpo_model_unsloth_ddp.py
CHANGED
|
@@ -92,7 +92,7 @@ else:
|
|
| 92 |
project_path = Path(os.path.abspath("../../../"))
|
| 93 |
temp_directory = Path("/root/autodl-tmp/OpenMiniMind/temp")
|
| 94 |
|
| 95 |
-
import unsloth
|
| 96 |
from datasets import load_dataset
|
| 97 |
from trl import GRPOConfig, GRPOTrainer
|
| 98 |
from unsloth import FastLanguageModel, is_bf16_supported
|
|
|
|
| 92 |
project_path = Path(os.path.abspath("../../../"))
|
| 93 |
temp_directory = Path("/root/autodl-tmp/OpenMiniMind/temp")
|
| 94 |
|
| 95 |
+
# import unsloth
|
| 96 |
from datasets import load_dataset
|
| 97 |
from trl import GRPOConfig, GRPOTrainer
|
| 98 |
from unsloth import FastLanguageModel, is_bf16_supported
|