- entklei.py +1 -1
- requirements.txt +2 -3
entklei.py
CHANGED
|
@@ -127,7 +127,7 @@ base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
|
|
| 127 |
base_pipe = base_pipe.to("cuda")
|
| 128 |
base_pipe.enable_model_cpu_offload()
|
| 129 |
base_pipe.safety_checker = None
|
| 130 |
-
base_pipe.enable_xformers_memory_efficient_attention()
|
| 131 |
|
| 132 |
pipe_with_tit_slider = _.clone_deep(base_pipe)
|
| 133 |
|
|
|
|
| 127 |
base_pipe = base_pipe.to("cuda")
|
| 128 |
base_pipe.enable_model_cpu_offload()
|
| 129 |
base_pipe.safety_checker = None
|
| 130 |
+
#base_pipe.enable_xformers_memory_efficient_attention()
|
| 131 |
|
| 132 |
pipe_with_tit_slider = _.clone_deep(base_pipe)
|
| 133 |
|
requirements.txt
CHANGED
|
@@ -3,12 +3,10 @@ torch==1.13.1+cu117
|
|
| 3 |
torchvision==0.14.1+cu117
|
| 4 |
torchaudio==0.13.1
|
| 5 |
accelerate
|
| 6 |
-
matplotlib
|
| 7 |
addict
|
| 8 |
diffusers
|
| 9 |
gradio==3.36.1
|
| 10 |
huggingface_hub
|
| 11 |
-
matplotlib
|
| 12 |
numpy
|
| 13 |
onnxruntime
|
| 14 |
opencv_python
|
|
@@ -51,4 +49,5 @@ tencentcloud-sdk-python
|
|
| 51 |
pydash
|
| 52 |
boto3
|
| 53 |
ninja # (Optional) Makes the build much faster
|
| 54 |
-
# git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
|
|
|
|
|
|
| 3 |
torchvision==0.14.1+cu117
|
| 4 |
torchaudio==0.13.1
|
| 5 |
accelerate
|
|
|
|
| 6 |
addict
|
| 7 |
diffusers
|
| 8 |
gradio==3.36.1
|
| 9 |
huggingface_hub
|
|
|
|
| 10 |
numpy
|
| 11 |
onnxruntime
|
| 12 |
opencv_python
|
|
|
|
| 49 |
pydash
|
| 50 |
boto3
|
| 51 |
ninja # (Optional) Makes the build much faster
|
| 52 |
+
# git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
|
| 53 |
+
matplotlib
|