Spaces:
Running
Running
Commit
·
3b7a372
1
Parent(s):
ad1b63f
update
Browse files- README.md +3 -1
- diffusers +1 -1
- requirements.txt +3 -1
README.md
CHANGED
@@ -14,6 +14,8 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
|
|
14 |
|
15 |
commands:
|
16 |
|
|
|
|
|
17 |
accelerate launch \
|
18 |
--deepspeed_config_file ds_config.json \
|
19 |
diffusers/examples/dreambooth/train_dreambooth.py \
|
@@ -35,7 +37,7 @@ accelerate launch \
|
|
35 |
--report_to="tensorboard" \
|
36 |
--logging_dir="./nyc-ad-model/logs"
|
37 |
|
38 |
-
|
39 |
|
40 |
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
41 |
|
|
|
14 |
|
15 |
commands:
|
16 |
|
17 |
+
pip install git+https://github.com/huggingface/diffusers
|
18 |
+
|
19 |
accelerate launch \
|
20 |
--deepspeed_config_file ds_config.json \
|
21 |
diffusers/examples/dreambooth/train_dreambooth.py \
|
|
|
37 |
--report_to="tensorboard" \
|
38 |
--logging_dir="./nyc-ad-model/logs"
|
39 |
|
40 |
+
|
41 |
|
42 |
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
43 |
|
diffusers
CHANGED
@@ -1 +1 @@
|
|
1 |
-
Subproject commit
|
|
|
1 |
+
Subproject commit 92fe689f06bcec27c4f48cb90574c2b9c42c643b
|
requirements.txt
CHANGED
@@ -4,4 +4,6 @@ invisible_watermark
|
|
4 |
torch
|
5 |
transformers
|
6 |
xformers
|
7 |
-
torchvision
|
|
|
|
|
|
4 |
torch
|
5 |
transformers
|
6 |
xformers
|
7 |
+
torchvision
|
8 |
+
flickrapi
|
9 |
+
requests
|