diff --git a/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/llava.egg-info/requires.txt b/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/llava.egg-info/requires.txt index 35197ff..1cc51d5 100644 --- a/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/llava.egg-info/requires.txt +++ b/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/llava.egg-info/requires.txt @@ -12,7 +12,7 @@ uvicorn wandb shortuuid httpx==0.24.0 -deepspeed==0.9.5 +deepspeed==0.15.1 peft==0.4.0 transformers==4.38.0 accelerate==0.21.0 diff --git a/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/pyproject.toml b/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/pyproject.toml index 3388c27..d47650a 100644 --- a/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/pyproject.toml +++ b/modules/advanced-zero-click-deployment-interface/FlowSteering/llava/pyproject.toml @@ -17,7 +17,7 @@ dependencies = [ "requests", "sentencepiece", "tokenizers>=0.12.1", "torch", "torchvision", "uvicorn", "wandb", "shortuuid", "httpx==0.24.0", - "deepspeed==0.9.5", + "deepspeed==0.15.1", "peft==0.4.0", "transformers==4.38.0", "accelerate==0.21.0",