Skip to content

Commit 113de54

Browse files
committed
disclaimer
1 parent 564a7a2 commit 113de54

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

roboflow/util/model_processor.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,14 +115,18 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
115115
except ImportError:
116116
raise RuntimeError(
117117
"The ultralytics python package is required to deploy yolov12"
118-
" models. Please install it with `pip install ultralytics`"
118+
" models. Please install it from `https://github.com/sunsmarterjie/yolov12`"
119119
)
120120

121121
print(
122-
"Note: Model must be trained using ultralytics from https://github.com/sunsmarterjie/yolov12 "
123-
"or through the Roboflow platform"
122+
"\n!!! ATTENTION !!!\n"
123+
"Model must be trained and uploaded using ultralytics from https://github.com/sunsmarterjie/yolov12\n"
124+
"or through the Roboflow platform\n"
125+
"!!! ATTENTION !!!\n"
124126
)
125127

128+
print_warn_for_wrong_dependencies_versions([("ultralytics", "==", "8.3.63")], ask_to_continue=True)
129+
126130
model = torch.load(os.path.join(model_path, filename), weights_only=False)
127131

128132
model_instance = model["model"] if "model" in model and model["model"] is not None else model["ema"]

0 commit comments

Comments
 (0)