From 9dcba3897810499786d1fb4b4bd8d41ef595a130 Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Fri, 24 Mar 2023 19:07:28 +0000 Subject: [PATCH] Pin transformers to a working Llama-compatible version --- environments/huggingface.yml | 2 +- environments/rocm.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 42dda9c3..6807627e 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -30,7 +30,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - git+https://github.com/zphang/transformers@llama_push + - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24 - git+https://github.com/huggingface/datasets@test-pandas-2.0.0rc - huggingface_hub==0.12.1 - safetensors diff --git a/environments/rocm.yml b/environments/rocm.yml index 43fd331f..a1d3d8b0 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -29,7 +29,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - transformers==4.25.1 + - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24 - huggingface_hub==0.12.1 - safetensors - accelerate