update environment variable's value
Browse files- app.py +4 -0
- trellis/pipelines/trellis_image_to_3d.py +4 -0
app.py
CHANGED
@@ -3,6 +3,10 @@ import spaces
|
|
3 |
from gradio_litmodel3d import LitModel3D
|
4 |
import io
|
5 |
import os
|
|
|
|
|
|
|
|
|
6 |
import shutil
|
7 |
os.environ['SPCONV_ALGO'] = 'native'
|
8 |
from typing import *
|
|
|
3 |
from gradio_litmodel3d import LitModel3D
|
4 |
import io
|
5 |
import os
|
6 |
+
import os
|
7 |
+
os.environ["OMP_NUM_THREADS"] = "3500"
|
8 |
+
|
9 |
+
|
10 |
import shutil
|
11 |
os.environ['SPCONV_ALGO'] = 'native'
|
12 |
from typing import *
|
trellis/pipelines/trellis_image_to_3d.py
CHANGED
@@ -8,6 +8,10 @@ from tqdm import tqdm
|
|
8 |
from easydict import EasyDict as edict
|
9 |
from torchvision import transforms
|
10 |
from PIL import Image
|
|
|
|
|
|
|
|
|
11 |
import rembg
|
12 |
from .base import Pipeline
|
13 |
from . import samplers
|
|
|
8 |
from easydict import EasyDict as edict
|
9 |
from torchvision import transforms
|
10 |
from PIL import Image
|
11 |
+
import os
|
12 |
+
|
13 |
+
os.environ["OMP_NUM_THREADS"] = "3500"
|
14 |
+
|
15 |
import rembg
|
16 |
from .base import Pipeline
|
17 |
from . import samplers
|