Update readme.md

diyiiyiii · Mar 25, 2022 · c9c1127 · c9c1127
1 parent ac292d6
commit c9c1127
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/readme.md b/readme.md
@@ -27,7 +27,7 @@ The overall pipeline of our StyTr^2 framework. We split the content and style im
 * tqdm  <br> 
 
 ### Testing 
-Pretrained models: [vgg-model](https://drive.google.com/file/d/1BinnwM5AmIcVubr16tPTqxMjUCE8iu5M/view?usp=sharing),  decoder[Coming SOON], Transformer_module [Coming SOON]   <br> 
+Pretrained models: [vgg-model](https://drive.google.com/file/d/1BinnwM5AmIcVubr16tPTqxMjUCE8iu5M/view?usp=sharing),  [vit_embedding](https://drive.google.com/file/d/1C3xzTOWx8dUXXybxZwmjijZN8SrC3e4B/view?usp=sharing)  [decoder](https://drive.google.com/file/d/1fIIVMTA_tPuaAAFtqizr6sd1XV7CX6F9/view?usp=sharing), Transformer_module [Coming SOON]   <br> 
 Please download them and put them into the floder  ./experiments/  <br> 
 ```
 python test.py  --content_dir input/content/ --style_dir input/style/    --output out