pillow_heif

app.py

@@ -11,6 +11,10 @@ from diffusers import (
     StableDiffusionXLPipeline,
 )
 from ip_adapter import StyleStudio_Adapter
+import pillow_heif
+
+pillow_heif.register_heif_opener()
+pillow_heif.register_avif_opener()
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
@@ -162,6 +166,7 @@ Advanced usage:<br>
 2. Choose different guidance and steps.
 3. Set the timing for the Teacher Model's participation.
 4. Feel free to discontinue using the Cross-Modal AdaIN and the Teacher Model for result comparison.
+5. ⏰ We recommend uploading images in <b>JPEG</b> or <b>PNG</b> format.
 """
 
 article = r"""
@@ -171,13 +176,22 @@ article = r"""
 1. As the value of end_fusion <b>increases</b>, the style gradually diminishes. 
 Therefore, it is suggested to set end_fusion to be between <b>1/5 and 1/3</b> of the number of inference steps (num inference steps).
 2. If you want to experience style-based CFG, see the details on the <a href="https://github.com/Westlake-AGI-Lab/StyleStudio">GitHub repo</a>.
+3. For the purpose of comparing various results, it is possible to do so by switching off the cross-modal AdaIN and the Teacher Model.
 
 ---
 📝 **Citation**
 <br>
 If our work is helpful for your research or applications, please cite us via:
 ```bibtex
-
+@misc{lei2024stylestudiotextdrivenstyletransfer,
+        title={StyleStudio: Text-Driven Style Transfer with Selective Control of Style Elements}, 
+        author={Mingkun Lei and Xue Song and Beier Zhu and Hao Wang and Chi Zhang},
+        year={2024},
+        eprint={2412.08503},
+        archivePrefix={arXiv},
+        primaryClass={cs.CV},
+        url={https://arxiv.org/abs/2412.08503}, 
+      }
 ```
 📧 **Contact**
 <br>

requirements.txt

@@ -15,4 +15,5 @@ insightface
 gradio
 controlnet_aux
 gdown
-peft
+peft
+pillow_heif