Skip to content

Commit 2e5e4e5

Browse files
committed
fixed import bug
1 parent e0454be commit 2e5e4e5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

apps/grpo/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@
2121
from forge.controller.actor import ForgeActor
2222
from forge.controller.service import ServiceConfig, shutdown_service, spawn_service
2323
from forge.data.rewards import MathReward, ThinkingReward
24+
from forge.data.utils import exclude_service
2425
from forge.util.metric_logging import get_metric_logger
2526
from monarch.actor import endpoint
2627
from omegaconf import DictConfig
27-
from src.forge.data.utils import exclude_service
2828
from torch import nn
2929
from transformers import AutoModelForCausalLM
3030
from vllm.transformers_utils.tokenizer import get_tokenizer

0 commit comments

Comments
 (0)