Skip to content

Commit

Permalink
add requirements (#231)
Browse files Browse the repository at this point in the history
add requirements to make some CI tests work
  • Loading branch information
ZefanW authored Feb 9, 2025
1 parent e842b73 commit 577a341
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 2 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,5 @@ transformers<4.48
vllm<=0.6.3
wandb
liger-kernel
pylatexenc
pyext
1 change: 0 additions & 1 deletion verl/workers/fsdp_workers.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@
import torch
import torch.distributed
from torch.distributed.device_mesh import init_device_mesh
import verl.utils.hdfs_io as hdfs_io
import verl.utils.torch_functional as verl_F
from omegaconf import DictConfig, open_dict
from verl import DataProto
Expand Down

0 comments on commit 577a341

Please sign in to comment.