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