XiangpengYang commited on
Commit
79634cc
·
1 Parent(s): 44e02db
README.md CHANGED
@@ -10,8 +10,8 @@ Our method is tested using cuda12.1, fp16 of accelerator and xformers on a singl
10
 
11
  ```bash
12
  # Step 1: Create and activate Conda environment
13
- conda create -n st-modulator python==3.10
14
- conda activate st-modulator
15
 
16
  # Step 2: Install PyTorch, CUDA and Xformers
17
  conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
@@ -36,13 +36,13 @@ gdown https://drive.google.com/file/d/1dzdvLnXWeMFR3CE2Ew0Bs06vyFSvnGXA/view?usp
36
  tar -zxvf videograin_data.tar.gz
37
  ```
38
 
39
- ## 🔥 ST-Modulator Editing
40
 
41
  You could reproduce multi-grained editing results in our teaser by running:
42
 
43
  ```bash
44
  bash test.sh
45
- #or accelerate launch test.py --config config/run_two_man.yaml
46
  ```
47
 
48
  <details><summary>The result is saved at `./result` . (Click for directory structure) </summary>
 
10
 
11
  ```bash
12
  # Step 1: Create and activate Conda environment
13
+ conda create -n videograin python==3.10
14
+ conda activate videograin
15
 
16
  # Step 2: Install PyTorch, CUDA and Xformers
17
  conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
 
36
  tar -zxvf videograin_data.tar.gz
37
  ```
38
 
39
+ ## 🔥 VideoGrain Editing
40
 
41
  You could reproduce multi-grained editing results in our teaser by running:
42
 
43
  ```bash
44
  bash test.sh
45
+ #or accelerate launch test.py --config config/instance_level/running_two_man/running_3cls_polar_spider_vis_weight.yaml
46
  ```
47
 
48
  <details><summary>The result is saved at `./result` . (Click for directory structure) </summary>
config/instance_level/running_two_man/{running_3cls_vis_cross_attn.yaml → running_3cls_polar_spider_vis_weight.yaml} RENAMED
File without changes