Upload folder using huggingface_hub
Browse files- chute_config.yml +2 -1
- miner.py +1 -1
- weights.onnx +2 -2
chute_config.yml
CHANGED
|
@@ -2,12 +2,13 @@ Image:
|
|
| 2 |
from_base: parachutes/python:3.12
|
| 3 |
run_command:
|
| 4 |
- pip install --upgrade setuptools wheel
|
| 5 |
-
- pip install 'numpy>=1.23' 'onnxruntime-gpu=
|
| 6 |
- pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
|
| 7 |
|
| 8 |
NodeSelector:
|
| 9 |
gpu_count: 1
|
| 10 |
min_vram_gb_per_gpu: 16
|
|
|
|
| 11 |
include:
|
| 12 |
- pro_6000
|
| 13 |
|
|
|
|
| 2 |
from_base: parachutes/python:3.12
|
| 3 |
run_command:
|
| 4 |
- pip install --upgrade setuptools wheel
|
| 5 |
+
- pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
|
| 6 |
- pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
|
| 7 |
|
| 8 |
NodeSelector:
|
| 9 |
gpu_count: 1
|
| 10 |
min_vram_gb_per_gpu: 16
|
| 11 |
+
max_hourly_price_per_gpu: 2
|
| 12 |
include:
|
| 13 |
- pro_6000
|
| 14 |
|
miner.py
CHANGED
|
@@ -50,7 +50,7 @@ class Miner:
|
|
| 50 |
max_det = 150
|
| 51 |
|
| 52 |
_conf_thres_array = np.array(
|
| 53 |
-
[0.38, 0.
|
| 54 |
)
|
| 55 |
_bonus_array = np.array(
|
| 56 |
[0.2, 0.25, 0.12, 0.09, 0.21, 0.06], dtype=np.float32,
|
|
|
|
| 50 |
max_det = 150
|
| 51 |
|
| 52 |
_conf_thres_array = np.array(
|
| 53 |
+
[0.38, 0.38, 0.22, 0.12, 0.33, 0.20], dtype=np.float32,
|
| 54 |
)
|
| 55 |
_bonus_array = np.array(
|
| 56 |
[0.2, 0.25, 0.12, 0.09, 0.21, 0.06], dtype=np.float32,
|
weights.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f0c937da3ee892f586a42b3a808d90f44e5c0f1ccc51c19d4889f5536a2b816
|
| 3 |
+
size 9844493
|