Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
Β·
bad2d4f
1
Parent(s):
f78f9ee
update
Browse files- app.py +1 -1
- embodied_gen/models/text_model.py +1 -0
- embodied_gen/models/texture_model.py +1 -2
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/
|
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 |
)
|