diff --git a/Dockerfile b/Dockerfile index 8fac8e4..7cb9553 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,7 +25,7 @@ RUN pip install --upgrade pip build COPY pyproject.toml /home/user/pyproject.toml COPY src/aihero /home/user/src/aihero -RUN pip install --no-cache-dir . +RUN pip install . # Run launch.py when the container launches COPY launch.py /home/user/ diff --git a/requirements-dev.txt b/requirements-dev.txt deleted file mode 100644 index b98adb7..0000000 --- a/requirements-dev.txt +++ /dev/null @@ -1,9 +0,0 @@ -pytest>=6.2.5 -black>=22.3 -mypy>=0.910 -ruff>=0.0.79 -blacken-docs>=1.11.0 -pyupgrade>=2.29.1 -detect-secrets>=1.2.0 -tomli>=1.2.3 -pre-commit>=2.17.0 diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index 41de1d9..0000000 --- a/requirements.txt +++ /dev/null @@ -1,18 +0,0 @@ -accelerate==0.25.0 -aihero-research-config -bitsandbytes==0.41.3.post2 -datasets==2.14.6 -einops==0.7.0 -fire==0.5.0 -minio==7.2.0 -numpy==1.26.4 -peft==0.7.1 -pydantic-settings==2.0.3 -python-dotenv==1.0.1 -PyYAML==6.0.1 -scikit-learn==1.4.0 -scipy==1.11.3 -transformers==4.36.1 -trl==0.7.7 -types-PyYAML==6.0.12.12 -wandb==0.15.12