goodmodeler commited on
Commit
4642679
·
1 Parent(s): b4b53f3
Files changed (2) hide show
  1. README.md +1 -2
  2. requirements.txt +2 -1
README.md CHANGED
@@ -33,7 +33,6 @@ accelerate launch \
33
  --checkpointing_steps=100 \
34
  --checkpoints_total_limit=1 \
35
  --report_to="tensorboard" \
36
- --logging_dir="./nyc-ad-model/logs" \
37
- --enable_xformers_memory_efficient_attention
38
 
39
  git clone https://github.com/huggingface/diffusers.git
 
33
  --checkpointing_steps=100 \
34
  --checkpoints_total_limit=1 \
35
  --report_to="tensorboard" \
36
+ --logging_dir="./nyc-ad-model/logs"
 
37
 
38
  git clone https://github.com/huggingface/diffusers.git
requirements.txt CHANGED
@@ -3,4 +3,5 @@ diffusers
3
  invisible_watermark
4
  torch
5
  transformers
6
- xformers
 
 
3
  invisible_watermark
4
  torch
5
  transformers
6
+ xformers
7
+ torchvision