From 9eb161def1e73ae81f4aaa674e1f74b7edd10397 Mon Sep 17 00:00:00 2001
From: Mylo <36931363+gitmylo@users.noreply.github.com>
Date: Fri, 16 Aug 2024 12:24:10 +0200
Subject: [PATCH] Update some requirement versions.

Upgrade huggingface-hub from 0.23.0 to 0.23.2
Lock transformers from latest to 4.44.0 (latest as the time of committing)
This should prevent some re-installs of packages, without doing a workaround.

This commit is related to the issues: #231, #236
---
 setup_tools/magicinstaller/requirements/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup_tools/magicinstaller/requirements/__init__.py b/setup_tools/magicinstaller/requirements/__init__.py
index ae67d58..58c8366 100644
--- a/setup_tools/magicinstaller/requirements/__init__.py
+++ b/setup_tools/magicinstaller/requirements/__init__.py
@@ -22,10 +22,10 @@
 
     Torch(),
 
-    Transformers(),
+    SimpleRequirementInit('transformers', CompareAction.EQ, '4.44.0'),
     diffusers(),  # This one's a function
     SimpleRequirementInit('gradio', CompareAction.EQ, '3.49.0'),
-    SimpleRequirementInit('huggingface-hub', CompareAction.EQ, '0.23.0'),
+    SimpleRequirementInit('huggingface-hub', CompareAction.EQ, '0.23.2'),
     AudioToNumpy(),
 
     Bark(),