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 65affa2 commit 2a129cbCopy full SHA for 2a129cb
verl/utils/transferqueue_utils.py
@@ -20,9 +20,12 @@
20
import numpy as np
21
import torch
22
from tensordict import NonTensorData, NonTensorStack, TensorDict
23
-from transfer_queue import AsyncTransferQueueClient, BatchMeta
24
25
-from verl.experimental.transfer_queue import ZMQServerInfo
+from verl.experimental.transfer_queue import (
+ AsyncTransferQueueClient,
26
+ BatchMeta,
27
+ ZMQServerInfo,
28
+)
29
from verl.protocol import DataProto
30
31
# _TRANSFER_QUEUE_CONTROLLER_INFOS = None
0 commit comments