Skip to content

Commit 20f59bf

Browse files
fix: fix compute_logprobs_parallel import
1 parent 3d4342b commit 20f59bf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/forge/actors/reference_model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@
1717
from forge.controller import ForgeActor
1818
from forge.observability.metrics import record_metric, Reduce
1919
from forge.observability.perf_tracker import Tracer
20-
from forge.util.ops import compute_logprobs
21-
from forge.util.parallel_logprobs import compute_logprobs_parallel
20+
from forge.util.ops import compute_logprobs, compute_logprobs_parallel
2221
from monarch.actor import current_rank, current_size, endpoint
2322
from torch.distributed.tensor import DTensor
2423

0 commit comments

Comments
 (0)