- entklei.py +1 -1
- requirements.txt +1 -1
entklei.py
CHANGED
|
@@ -129,7 +129,7 @@ base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
|
|
| 129 |
base_pipe = base_pipe.to("cuda")
|
| 130 |
base_pipe.enable_model_cpu_offload()
|
| 131 |
base_pipe.safety_checker = None
|
| 132 |
-
|
| 133 |
|
| 134 |
pipe_with_tit_slider = _.clone_deep(base_pipe)
|
| 135 |
|
|
|
|
| 129 |
base_pipe = base_pipe.to("cuda")
|
| 130 |
base_pipe.enable_model_cpu_offload()
|
| 131 |
base_pipe.safety_checker = None
|
| 132 |
+
base_pipe.enable_xformers_memory_efficient_attention()
|
| 133 |
|
| 134 |
pipe_with_tit_slider = _.clone_deep(base_pipe)
|
| 135 |
|
requirements.txt
CHANGED
|
@@ -46,7 +46,7 @@ tencentcloud-sdk-python
|
|
| 46 |
pydash
|
| 47 |
boto3
|
| 48 |
ninja # (Optional) Makes the build much faster
|
| 49 |
-
|
| 50 |
git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
|
| 51 |
git+https://github.com/huggingface/peft.git@main#egg=peft
|
| 52 |
git+https://github.com/huggingface/transformers.git@main#egg=transformers
|
|
|
|
| 46 |
pydash
|
| 47 |
boto3
|
| 48 |
ninja # (Optional) Makes the build much faster
|
| 49 |
+
git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
|
| 50 |
git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
|
| 51 |
git+https://github.com/huggingface/peft.git@main#egg=peft
|
| 52 |
git+https://github.com/huggingface/transformers.git@main#egg=transformers
|