From 198e2920d27459225eafae7779fab65e4d4d9c06 Mon Sep 17 00:00:00 2001 From: Divided by Zer0 Date: Fri, 16 Sep 2022 00:52:15 +0200 Subject: [PATCH] loguru dependency in all environments --- environments/huggingface.yml | 2 +- environments/rocm-finetune.yml | 1 + environments/rocm.yml | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) 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.*