JiantaoLin
commited on
Commit
·
51e4765
1
Parent(s):
73a0140
new
Browse files- pipeline/utils.py +1 -0
pipeline/utils.py
CHANGED
|
@@ -34,6 +34,7 @@ TMP_DIR = './outputs/tmp'
|
|
| 34 |
|
| 35 |
os.makedirs(TMP_DIR, exist_ok=True)
|
| 36 |
|
|
|
|
| 37 |
def lrm_reconstruct(model, infer_config, images,
|
| 38 |
name='', export_texmap=False,
|
| 39 |
input_camera_type='zero123',
|
|
|
|
| 34 |
|
| 35 |
os.makedirs(TMP_DIR, exist_ok=True)
|
| 36 |
|
| 37 |
+
@torch.no_grad()
|
| 38 |
def lrm_reconstruct(model, infer_config, images,
|
| 39 |
name='', export_texmap=False,
|
| 40 |
input_camera_type='zero123',
|