From 834fd3f55377c0c25636fd76d035fd6e28b3fa8b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Kn=C3=B6rig?=
 <aknoerig@users.noreply.github.com>
Date: Mon, 20 Jan 2025 16:03:49 +0100
Subject: [PATCH] reduce dependencies to minimum

---
 requirements.txt |  9 ++++++++-
 setup.py         | 11 +----------
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index f2edf38..434bc97 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,10 @@
+# Development requirements
+torch
+torchvision
+torchaudio
 transformers
 diffusers
-lark
\ No newline at end of file
+lark
+optimum-quanto
+safetensors
+torchao
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 6de1e8f..6a1a5aa 100644
--- a/setup.py
+++ b/setup.py
@@ -11,19 +11,10 @@
     , url              = 'https://github.com/xhinker/sd_embed'
     , keywords         = 'diffusers stable-diffusion embedding'
     , install_requires = [
-        'optimum-quanto'
-        , 'torch'
-        , 'torchvision'
-        , 'torchaudio'
-        , 'sentencepiece'
-        , 'accelerate'
-        , 'peft'
+        'torch'
         , 'transformers'
         , 'diffusers'
         , 'lark'
-        , 'protobuf'
-        , 'ipykernel'
-        , 'ipywidgets'
         , 'safetensors'
     ]
     , include_package_data=True