Skip to content

Commit 95d95df

Browse files
authored
📦️ add weights usage
1 parent d4cc602 commit 95d95df

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)