Skip to content

Commit 4cd92e3

Browse files
committed
remove checkpoint
1 parent cc7f697 commit 4cd92e3

File tree

3 files changed

+1
-3
lines changed

3 files changed

+1
-3
lines changed

src/twinkle/infra/_ray/resource_manager.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,6 @@ def __init__(self,
7171
self.min_node_idx = 0
7272
self.nnodes = math.ceil(cpu_proc_count / ncpu_proc_per_node)
7373

74-
# breakpoint()
7574
self.nodes = []
7675
for node in ray.nodes():
7776
# get available nodes

src/twinkle/loss/grpo.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,6 @@ def __call__(
305305
Returns:
306306
loss: Scalar loss value
307307
"""
308-
# breakpoint()
309308
import torch
310309
labels = inputs.get('labels')
311310
assert labels is not None, "inputs must contain 'labels'"

src/twinkle/server/tinker/common/transformers_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def forward_backward(self, *, inputs: List[types.Datum], adapter_name: str, loss
114114

115115
# Convert Datum to InputFeature
116116
input_features = datum_to_input_feature(inputs, template)
117-
# breakpoint()
117+
118118
# Forward pass
119119
outputs = super().forward(inputs=input_features, adapter_name=adapter_name, **kwargs)
120120

0 commit comments

Comments
 (0)