MohamedRashad commited on
Commit
830753e
1 Parent(s): 853604a

Add nvdiffrast import to mesh_renderer.py for rendering enhancements

Browse files
Files changed (1) hide show
  1. trellis/renderers/mesh_renderer.py +1 -0
trellis/renderers/mesh_renderer.py CHANGED
@@ -6,6 +6,7 @@
6
  # distribution of this software and related documentation without an express
7
  # license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
8
  import torch
 
9
  from easydict import EasyDict as edict
10
  from ..representations.mesh import MeshExtractResult
11
  import torch.nn.functional as F
 
6
  # distribution of this software and related documentation without an express
7
  # license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
8
  import torch
9
+ import nvdiffrast.torch as dr
10
  from easydict import EasyDict as edict
11
  from ..representations.mesh import MeshExtractResult
12
  import torch.nn.functional as F