diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 3daec6ae..7abceefa 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -19,10 +19,10 @@ dependencies: - protobuf - marshmallow>=3.13 - apispec-webframeworks + - loguru - pip: - flask-cloudflared - flask-ngrok - lupa==1.10 - transformers>=4.20.1 - accelerate - - loguru \ No newline at end of file diff --git a/environments/rocm-finetune.yml b/environments/rocm-finetune.yml index 05d7a2c1..fc56eb4f 100644 --- a/environments/rocm-finetune.yml +++ b/environments/rocm-finetune.yml @@ -14,6 +14,7 @@ dependencies: - git=2.35.1 - marshmallow>=3.13 - apispec-webframeworks + - loguru - pip: - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html - torch diff --git a/environments/rocm.yml b/environments/rocm.yml index 142d4230..fef38892 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -16,6 +16,7 @@ dependencies: - protobuf - marshmallow>=3.13 - apispec-webframeworks + - loguru - pip: - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html - torch==1.10.*