xinjie.wang commited on
Commit
bad2d4f
Β·
1 Parent(s): f78f9ee
app.py CHANGED
@@ -48,7 +48,7 @@ with gr.Blocks(delete_cache=(43200, 43200), theme=custom_theme) as demo:
48
  <a href="https://horizonrobotics.github.io/robot_lab/embodied_gen/index.html">
49
  <img alt="🌐 Project Page" src="https://img.shields.io/badge/🌐-Project_Page-blue">
50
  </a>
51
- <a href="https://arxiv.org/abs/xxxx.xxxxx">
52
  <img alt="πŸ“„ arXiv" src="https://img.shields.io/badge/πŸ“„-arXiv-b31b1b">
53
  </a>
54
  <a href="https://github.com/HorizonRobotics/EmbodiedGen">
 
48
  <a href="https://horizonrobotics.github.io/robot_lab/embodied_gen/index.html">
49
  <img alt="🌐 Project Page" src="https://img.shields.io/badge/🌐-Project_Page-blue">
50
  </a>
51
+ <a href="https://arxiv.org/abs/2506.10600">
52
  <img alt="πŸ“„ arXiv" src="https://img.shields.io/badge/πŸ“„-arXiv-b31b1b">
53
  </a>
54
  <a href="https://github.com/HorizonRobotics/EmbodiedGen">
embodied_gen/models/text_model.py CHANGED
@@ -19,6 +19,7 @@ import logging
19
  import os
20
  import random
21
  import subprocess
 
22
  import numpy as np
23
  import torch
24
  from diffusers import (
 
19
  import os
20
  import random
21
  import subprocess
22
+
23
  import numpy as np
24
  import torch
25
  from diffusers import (
embodied_gen/models/texture_model.py CHANGED
@@ -30,7 +30,6 @@ from kolors.pipelines.pipeline_controlnet_xl_kolors_img2img import (
30
  from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
31
  from embodied_gen.models.text_model import download_kolors_weights
32
 
33
-
34
  __all__ = [
35
  "build_texture_gen_pipe",
36
  ]
@@ -43,7 +42,7 @@ def build_texture_gen_pipe(
43
  device: str = "cuda",
44
  ) -> DiffusionPipeline:
45
  download_kolors_weights(f"{base_ckpt_dir}/Kolors")
46
-
47
  tokenizer = ChatGLMTokenizer.from_pretrained(
48
  f"{base_ckpt_dir}/Kolors/text_encoder"
49
  )
 
30
  from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
31
  from embodied_gen.models.text_model import download_kolors_weights
32
 
 
33
  __all__ = [
34
  "build_texture_gen_pipe",
35
  ]
 
42
  device: str = "cuda",
43
  ) -> DiffusionPipeline:
44
  download_kolors_weights(f"{base_ckpt_dir}/Kolors")
45
+
46
  tokenizer = ChatGLMTokenizer.from_pretrained(
47
  f"{base_ckpt_dir}/Kolors/text_encoder"
48
  )