Spaces:

luoxue-star
/

AniMerPlus

Running

luoxue-star commited on 25 days ago

Commit

abce00c

1 Parent(s): 399017c

Fix bug: remove cuda device

Files changed (1) hide show

amr/models/animerpp.py CHANGED Viewed

@@ -81,7 +81,7 @@ class AniMerPlusPlus(pl.LightningModule):
         # Only appling for AVES training
         self.aves_silouette_render = SilhouetteRenderer(size=self.cfg.MODEL.IMAGE_SIZE,
                                                         focal=self.cfg.AVES.get("FOCAL_LENGTH", 2167),
-                                                        device='cuda')
         self.automatic_optimization = False

         # Only appling for AVES training
         self.aves_silouette_render = SilhouetteRenderer(size=self.cfg.MODEL.IMAGE_SIZE,
                                                         focal=self.cfg.AVES.get("FOCAL_LENGTH", 2167),
+                                                        device='cpu')
         self.automatic_optimization = False