update weights link

wasserth · Jan 29, 2025 · 5d44ac3 · 5d44ac3
1 parent 88db8d9
commit 5d44ac3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/totalsegmentator/libs.py b/totalsegmentator/libs.py
@@ -240,7 +240,7 @@ def download_pretrained_weights(task_id):
         WEIGHTS_URL = url + "/v2.5.0-weights/Dataset527_breasts_1559subj.zip"
     elif task_id == 552:
         weights_path = config_dir / "Dataset552_ventricle_parts_38subj"
-        WEIGHTS_URL = url + "/v2.6.0-weights/Dataset552_ventricle_parts_38subj.zip"  # TODO: upload. does not exist yet.
+        WEIGHTS_URL = url + "/v2.5.0-weights/Dataset552_ventricle_parts_38subj.zip"
 
     # MR models
     elif task_id == 850: