Hugging Face's logo

Spaces:
czl
/
generative-data-augmentation-demo
Running on Zero

App Files Community
1
czl commited on
Commit
9b0248b
•
1 Parent(s): c5a3c44

update installation

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. requirements.txt +4 -1
README.md CHANGED
@@ -46,7 +46,7 @@ The user can augment an image by interpolating between two prompts, and specify
46
  ## Local Setup
47
 
48
  ```bash
49
- git clone https://huggingface.co/spaces/czl/generative-data-augmentation-demo
50
  cd generative-data-augmentation-demo
51
  # Setup the data directory structure as shown above
52
  conda create --name $env_name python=3.11.* # Replace $env_name with your environment name
 
46
  ## Local Setup
47
 
48
  ```bash
49
+ git clone https://github.com/zhulinchng/generative-data-augmentation-demo
50
  cd generative-data-augmentation-demo
51
  # Setup the data directory structure as shown above
52
  conda create --name $env_name python=3.11.* # Replace $env_name with your environment name
requirements.txt CHANGED
@@ -1,10 +1,13 @@
1
  accelerate
2
  diffusers
3
  invisible_watermark
 
4
  torch==2.1.2
 
5
  torchaudio==2.1.2
 
6
  torchvision==0.16.2
7
- transformers
8
  xformers==0.0.23.post1
9
  DeepCache
10
  pandas
 
1
  accelerate
2
  diffusers
3
  invisible_watermark
4
+ --index-url https://download.pytorch.org/whl/cu121
5
  torch==2.1.2
6
+ --index-url https://download.pytorch.org/whl/cu121
7
  torchaudio==2.1.2
8
+ --index-url https://download.pytorch.org/whl/cu121
9
  torchvision==0.16.2
10
+ transformers[torch]
11
  xformers==0.0.23.post1
12
  DeepCache
13
  pandas