diff --git a/docker/requirements-runtime.txt b/docker/requirements-runtime.txt index 5ab2b2a..43d3b3c 100644 --- a/docker/requirements-runtime.txt +++ b/docker/requirements-runtime.txt @@ -16,3 +16,4 @@ speedtest-cli tensorboard==2.12.0 wandb safetensors +prodigyopt diff --git a/windows_setup.cmd b/windows_setup.cmd index e6fe580..e81d9ff 100644 --- a/windows_setup.cmd +++ b/windows_setup.cmd @@ -23,6 +23,7 @@ pip install compel~=1.1.3 pip install dadaptation pip install safetensors pip install open-flamingo==2.0.0 +pip install prodigyopt python utils/get_yamls.py GOTO :eof