We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e0454be commit 2e5e4e5Copy full SHA for 2e5e4e5
apps/grpo/main.py
@@ -21,10 +21,10 @@
21
from forge.controller.actor import ForgeActor
22
from forge.controller.service import ServiceConfig, shutdown_service, spawn_service
23
from forge.data.rewards import MathReward, ThinkingReward
24
+from forge.data.utils import exclude_service
25
from forge.util.metric_logging import get_metric_logger
26
from monarch.actor import endpoint
27
from omegaconf import DictConfig
-from src.forge.data.utils import exclude_service
28
from torch import nn
29
from transformers import AutoModelForCausalLM
30
from vllm.transformers_utils.tokenizer import get_tokenizer
0 commit comments