Spaces:
mashroo
/
Running on Zero

YoussefAnso commited on
Commit
b2dc9cc
·
1 Parent(s): 2a4c225

Fix variable names in mesh export for CRM model to enhance clarity and maintain consistency with previous changes. Updated vertices and faces variable names for better readability.

Browse files
Files changed (1) hide show
  1. model/crm/model.py +1 -1
model/crm/model.py CHANGED
@@ -107,7 +107,7 @@ class CRM(nn.Module):
107
 
108
  # export the final mesh
109
  with torch.no_grad():
110
- mesh = trimesh.Trimesh(vertices=v_np, faces=f_np, process=False) # important, process=True leads to seg fault...
111
  mesh.export(f"{out_dir}.glb", file_type="glb")
112
 
113
  def export_mesh_wt_uv(self, ctx, data, out_dir, ind, device, res, tri_fea_2=None):
 
107
 
108
  # export the final mesh
109
  with torch.no_grad():
110
+ mesh = trimesh.Trimesh(vertices=verts, faces=faces, process=False) # important, process=True leads to seg fault...
111
  mesh.export(f"{out_dir}.glb", file_type="glb")
112
 
113
  def export_mesh_wt_uv(self, ctx, data, out_dir, ind, device, res, tri_fea_2=None):