From 577a341bf34e2436ad88f5e7427d50a497352a27 Mon Sep 17 00:00:00 2001 From: Zefan Wang Date: Sun, 9 Feb 2025 19:41:02 +0800 Subject: [PATCH] add requirements (#231) add requirements to make some CI tests work --- requirements.txt | 2 ++ verl/workers/fsdp_workers.py | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index caab0e2f..1e3e3b8b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,3 +15,5 @@ transformers<4.48 vllm<=0.6.3 wandb liger-kernel +pylatexenc +pyext \ No newline at end of file diff --git a/verl/workers/fsdp_workers.py b/verl/workers/fsdp_workers.py index 2965bd6d..ed96127d 100644 --- a/verl/workers/fsdp_workers.py +++ b/verl/workers/fsdp_workers.py @@ -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