update

app.py

@@ -58,7 +58,7 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
             <a href="https://horizonrobotics.github.io/robot_lab/embodied_gen/index.html">
                 <img alt="🌐 Project Page" src="https://img.shields.io/badge/🌐-Project_Page-blue">
             </a>
-            <a href="https://arxiv.org/abs/xxxx.xxxxx">
+            <a href="https://arxiv.org/abs/2506.10600">
                 <img alt="📄 arXiv" src="https://img.shields.io/badge/📄-arXiv-b31b1b">
             </a>
             <a href="https://github.com/HorizonRobotics/EmbodiedGen">

embodied_gen/models/text_model.py

@@ -19,6 +19,7 @@ import logging
 import os
 import random
 import subprocess
+
 import numpy as np
 import torch
 from diffusers import (

embodied_gen/models/texture_model.py

@@ -30,7 +30,6 @@ from kolors.pipelines.pipeline_controlnet_xl_kolors_img2img import (
 from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
 from embodied_gen.models.text_model import download_kolors_weights
 
-
 __all__ = [
     "build_texture_gen_pipe",
 ]
@@ -43,7 +42,7 @@ def build_texture_gen_pipe(
     device: str = "cuda",
 ) -> DiffusionPipeline:
     download_kolors_weights(f"{base_ckpt_dir}/Kolors")
-        
+
     tokenizer = ChatGLMTokenizer.from_pretrained(
         f"{base_ckpt_dir}/Kolors/text_encoder"
     )