Spaces:
Runtime error
Runtime error
fix requirements
Browse files- app.py +1 -1
- requirements.txt +3 -3
app.py
CHANGED
|
@@ -28,7 +28,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
| 28 |
)
|
| 29 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 30 |
pipe.enable_model_cpu_offload()
|
| 31 |
-
|
| 32 |
|
| 33 |
generator = torch.manual_seed(0)
|
| 34 |
|
|
|
|
| 28 |
)
|
| 29 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 30 |
pipe.enable_model_cpu_offload()
|
| 31 |
+
pipe.enable_xformers_memory_efficient_attention()
|
| 32 |
|
| 33 |
generator = torch.manual_seed(0)
|
| 34 |
|
requirements.txt
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
| 2 |
-
jax[
|
| 3 |
jaxlib
|
| 4 |
git+https://github.com/huggingface/diffusers@main
|
| 5 |
-
flax
|
| 6 |
opencv-python
|
| 7 |
transformers
|
| 8 |
-
|
| 9 |
pillow
|
| 10 |
triton
|
| 11 |
torch
|
| 12 |
torchvision
|
|
|
|
| 13 |
xformers==0.0.16
|
|
|
|
| 1 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
| 2 |
+
jax[cuda11_pip]
|
| 3 |
jaxlib
|
| 4 |
git+https://github.com/huggingface/diffusers@main
|
|
|
|
| 5 |
opencv-python
|
| 6 |
transformers
|
| 7 |
+
flax
|
| 8 |
pillow
|
| 9 |
triton
|
| 10 |
torch
|
| 11 |
torchvision
|
| 12 |
+
git+https://github.com/huggingface/accelerate
|
| 13 |
xformers==0.0.16
|