update

assets/example_texture/text_prompts.txt

@@ -4,8 +4,8 @@ assets/example_texture/meshes/robot_text.obj \ 举着牌子的红色写实风格
 assets/example_texture/meshes/robot_text.obj \ 举着牌子的彩色写实风格机器人，牌子上写着“World”的文字
 assets/example_texture/meshes/robot_text.obj \ 举着木质牌子的写实风格机甲3D模型，主体红色，牌子上写着“中国”的文字
 assets/example_texture/meshes/robot_text.obj \ 举着牌子的黄色写实风格机器人，牌子上画着爱心
-assets/example_texture/meshes/horse.obj \ 具有古典质感、鬃毛飞扬的棕色马首，黑色眼睛
-assets/example_texture/meshes/horse.obj \ 具有古典质感、鬃毛飞扬的栗色马首
+assets/example_texture/meshes/horse.obj \ 一匹棕色的马头，有着经典的纹理和飞扬的鬃毛，还有黑色的眼睛
+assets/example_texture/meshes/horse.obj \ 一匹灰色的马头，鬃毛飞扬，眼睛为棕色
 assets/example_texture/meshes/electric_drill.obj \ 电动手钻，有磨损细节，灰色红色相间磨砂质感哑光外壳
 assets/example_texture/meshes/electric_drill.obj \ 电动手钻，有磨损细节，黄蓝色哑光外壳带有磨砂质感
 assets/example_texture/meshes/chair.obj \ 具有大理石材质框架，坐垫为柔软青色织物纹理的旋转吧台椅
@@ -14,8 +14,6 @@ assets/example_texture/meshes/vase.obj \ 瓶身上绘有彩色飞鸟图案的花
 assets/example_texture/meshes/vase.obj \ 陶土花瓶
 assets/example_texture/meshes/globe.obj \ 木质底座的经纬线地球仪
 assets/example_texture/meshes/flashlight.obj \ 筒橡胶防滑纹路，红色磨砂质感的手电筒
-assets/example_texture/meshes/bed.obj \ 简约现代风格的双人床，铺着灰色被子，上面摆放着几个红色的抱枕
-assets/example_texture/meshes/bed.obj \ 北欧风格的双人床，浅灰色床单上面摆放着几个紫色的抱枕
 assets/example_texture/meshes/desk.obj \ 欧式风格木质梳妆台及配套凳子
 assets/example_texture/meshes/desk.obj \ 白色带抽屉和镜子的欧式风格梳妆台及配套凳子模型，添加金色装饰细节
 assets/example_texture/meshes/clock.obj \ 木质闹钟，黑色指针，显示时间为数字

embodied_gen/data/utils.py

@@ -139,7 +139,7 @@ class DiffrastRender(object):
         vertices: torch.Tensor,
         matrix: torch.Tensor,
     ) -> torch.Tensor:
-        verts_ones = torch.ones((len(vertices), 1)).to(vertices)
+        verts_ones = torch.ones((len(vertices), 1), device=vertices.device, dtype=vertices.dtype)
         verts_homo = torch.cat([vertices, verts_ones], dim=-1)
         trans_vertices = torch.matmul(verts_homo, matrix.permute(0, 2, 1))