Spaces:
Running
on
Zero
Running
on
Zero
MohamedRashad
commited on
Commit
•
830753e
1
Parent(s):
853604a
Add nvdiffrast import to mesh_renderer.py for rendering enhancements
Browse files
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
|