Adding HF files

requirements.txt

@@ -9,14 +9,15 @@
 # timm
 # OmegaConf
 
-# --extra-index-url https://download.pytorch.org/whl/cu116
-# torch==1.13.1+cu116
-# torchvision==0.14.1+cu116
+--extra-index-url https://download.pytorch.org/whl/cu116
+torch==1.13.1+cu116
+torchvision==0.14.1+cu116
 
 
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==1.13.1+cpu
-torchvision==0.14.1+cpu
+# --extra-index-url https://download.pytorch.org/whl/cpu
+# torch==1.13.1+cpu
+# torchvision==0.14.1+cpu
+
 pytorch-lightning
 smplx==0.1.28
 opencv-python

vendor/pyrender/pyrender/offscreen.py

@@ -127,27 +127,25 @@ class OffscreenRenderer(object):
         gc.collect()
 
     def _create(self):
-        # import os
-        # os.environ['DISPLAY'] = ':1'
-        # from pyrender.platforms.pyglet_platform import PygletPlatform
-        # self._platform = PygletPlatform(self.viewport_width,
-        #                                 self.viewport_height)
-        # if 'PYOPENGL_PLATFORM' not in os.environ:
-        # elif os.environ['PYOPENGL_PLATFORM'] == 'egl':
-        from pyrender.platforms import egl
-        device_id = int(os.environ.get('EGL_DEVICE_ID', '0'))
-        egl_device = egl.get_device_by_index(device_id)
-        self._platform = egl.EGLPlatform(self.viewport_width,
-                                            self.viewport_height,
-                                            device=egl_device)
-        # elif os.environ['PYOPENGL_PLATFORM'] == 'osmesa':
-        # from pyrender.platforms.osmesa import OSMesaPlatform
-        # self._platform = OSMesaPlatform(self.viewport_width,
-        #                                 self.viewport_height)
-        # else:
-        #     raise ValueError('Unsupported PyOpenGL platform: {}'.format(
-        #         os.environ['PYOPENGL_PLATFORM']
-        #     ))
+        if 'PYOPENGL_PLATFORM' not in os.environ:
+            from pyrender.platforms.pyglet_platform import PygletPlatform
+            self._platform = PygletPlatform(self.viewport_width,
+                                            self.viewport_height)
+        elif os.environ['PYOPENGL_PLATFORM'] == 'egl':
+            from pyrender.platforms import egl
+            device_id = int(os.environ.get('EGL_DEVICE_ID', '0'))
+            egl_device = egl.get_device_by_index(device_id)
+            self._platform = egl.EGLPlatform(self.viewport_width,
+                                             self.viewport_height,
+                                             device=egl_device)
+        elif os.environ['PYOPENGL_PLATFORM'] == 'osmesa':
+            from pyrender.platforms.osmesa import OSMesaPlatform
+            self._platform = OSMesaPlatform(self.viewport_width,
+                                            self.viewport_height)
+        else:
+            raise ValueError('Unsupported PyOpenGL platform: {}'.format(
+                os.environ['PYOPENGL_PLATFORM']
+            ))
         self._platform.init_context()
         self._platform.make_current()
         self._renderer = Renderer(self.viewport_width, self.viewport_height)