Upload folder using huggingface_hub
Browse files- config.yml +1 -1
- miner.py +1 -1
config.yml
CHANGED
|
@@ -18,7 +18,7 @@ NodeSelector:
|
|
| 18 |
- mi300x
|
| 19 |
|
| 20 |
Chute:
|
| 21 |
-
timeout_seconds:
|
| 22 |
concurrency: 4
|
| 23 |
max_instances: 5
|
| 24 |
scaling_threshold: 0.5
|
|
|
|
| 18 |
- mi300x
|
| 19 |
|
| 20 |
Chute:
|
| 21 |
+
timeout_seconds: 900
|
| 22 |
concurrency: 4
|
| 23 |
max_instances: 5
|
| 24 |
scaling_threshold: 0.5
|
miner.py
CHANGED
|
@@ -7,6 +7,7 @@ from numpy import ndarray
|
|
| 7 |
import numpy as np
|
| 8 |
from pydantic import BaseModel
|
| 9 |
import cv2
|
|
|
|
| 10 |
|
| 11 |
import onnxruntime as ort
|
| 12 |
from torchvision.ops import batched_nms
|
|
@@ -19,7 +20,6 @@ from utils import (
|
|
| 19 |
classify_teams_batch,
|
| 20 |
)
|
| 21 |
|
| 22 |
-
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
| 23 |
|
| 24 |
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
| 25 |
os.environ["OMP_NUM_THREADS"] = "16"
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
from pydantic import BaseModel
|
| 9 |
import cv2
|
| 10 |
+
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
| 11 |
|
| 12 |
import onnxruntime as ort
|
| 13 |
from torchvision.ops import batched_nms
|
|
|
|
| 20 |
classify_teams_batch,
|
| 21 |
)
|
| 22 |
|
|
|
|
| 23 |
|
| 24 |
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
| 25 |
os.environ["OMP_NUM_THREADS"] = "16"
|