File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -77,6 +77,8 @@ We also provide the labeling text dataset [here](data/mt_text_anno.json).
7777
7878## 📦 B. Usage
7979
80+ The pretrained weight is uploaded to [ Hugging Face] ( https://huggingface.co/Justin900/MAD ) 🤗.
81+
8082### B.1 Training a Model
8183
8284* With our model
@@ -117,7 +119,7 @@ python generate_translation.py \
117119 --source-label 0 \
118120 --target-label 1 \
119121 --num-process {NUM_PROCESS} \
120- --opts MODEL.PRETRAINED {MODEL_WEIGHTS}
122+ --opts MODEL.PRETRAINED Justin900/MAD
121123```
122124
123125### B.3 Makeup Transfer
@@ -146,7 +148,7 @@ python generate_transfer.py \
146148 --num-process {NUM_PROCESS} \
147149 --inpainting \
148150 --cam \
149- --opts MODEL.PRETRAINED {MODEL_WEIGHTS}
151+ --opts MODEL.PRETRAINED Justin900/MAD
150152```
151153
152154### B.4 Text Modification
@@ -166,7 +168,7 @@ python generate_text_editing.py \
166168 --source-root data/mtdataset/images \
167169 --source-list assets/text_editing.json \
168170 --num-process {NUM_PROCESS} \
169- --model-path {MODEL_WEIGHTS}
171+ --model-path Justin900/MAD
170172```
171173
172174## 🎨 C. Web UI (Beta)
You can’t perform that action at this time.
0 commit comments