Update monai/losses/perceptual.py

ericspod · coderabbitai[bot] · web-flow · commit 2156b8472349 · 2025-12-05T17:35:10.000Z
Co-authored-by: coderabbitai[bot] &lt;136622811+coderabbitai[bot]@users.noreply.github.com&gt;
Signed-off-by: Eric Kerfoot &lt;17726042+ericspod@users.noreply.github.com&gt;
diff --git a/monai/losses/perceptual.py b/monai/losses/perceptual.py
@@ -333,7 +333,7 @@ def __init__(self, net: str = "radimagenet_resnet50", verbose: bool = False, cac
             raise ValueError(
                 f"Invalid download model name '{net}'. Must be one of: {', '.join(HF_MONAI_MODELS)}."
         )
-        self.model = torch.hub.load("Project-MONAI/perceptual-models", model=net, verbose=verbose, cache_dir=cache_dir,
+        self.model = torch.hub.load("Project-MONAI/perceptual-models:main", model=net, verbose=verbose, cache_dir=cache_dir,
                                     trust_repo=True)
         self.eval()
 

Original file line number	Diff line number	Diff line change
`@@ -333,7 +333,7 @@ def __init__(self, net: str = "radimagenet_resnet50", verbose: bool = False, cac`
`333`	`333`	`raise ValueError(`
`334`	`334`	`f"Invalid download model name '{net}'. Must be one of: {', '.join(HF_MONAI_MODELS)}."`
`335`	`335`	`)`
`336`		`- self.model = torch.hub.load("Project-MONAI/perceptual-models", model=net, verbose=verbose, cache_dir=cache_dir,`
	`336`	`+ self.model = torch.hub.load("Project-MONAI/perceptual-models:main", model=net, verbose=verbose, cache_dir=cache_dir,`
`337`	`337`	`trust_repo=True)`
`338`	`338`	`self.eval()`
`339`	`339`