Spaces:

xinjjj
/

ImgRoboAssetGen

Running on Zero

xinjie.wang commited on 12 days ago

Commit

dc66801

1 Parent(s): be7eba9

update

Files changed (1) hide show

asset3d_gen/models/gs_model.py CHANGED Viewed

@@ -426,6 +426,7 @@ class GaussianOperator(GaussianBase):
             _features_dc=self._features_dc,
             _features_rest=self._features_rest,
             sh_degree=self.sh_degree,
         )
         return GaussianOperator(**gs_dict)

             _features_dc=self._features_dc,
             _features_rest=self._features_rest,
             sh_degree=self.sh_degree,
+            device=self.device,
         )
         return GaussianOperator(**gs_dict)