add a HF accelerate requirement (#1651)

2026-01-09 13:57:54 -05:00 · 2023-07-14 05:56:12 -07:00
parent 7cf19230e2
commit 5ec91143f5
2 changed files with 4 additions and 1 deletions
--- a/apps/stable_diffusion/shark_studio_imports.py
+++ b/apps/stable_diffusion/shark_studio_imports.py
@@ -24,6 +24,7 @@ datas += copy_metadata("Pillow")
 datas += copy_metadata("sentencepiece")
 datas += collect_data_files("tokenizers")
 datas += collect_data_files("tiktoken")
+datas += collect_data_files("accelerate")
 datas += collect_data_files("diffusers")
 datas += collect_data_files("transformers")
 datas += collect_data_files("pytorch_lightning")
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,9 @@ parameterized

 # Add transformers, diffusers and scipy since it most commonly used
 transformers
-diffusers==0.18.1
+diffusers
+#accelerate is now required for diffusers import from ckpt.
+accelerate
 scipy
 ftfy
 gradio