We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6cf0b5b commit baf8cf4Copy full SHA for baf8cf4
1 file changed
cellpose/vit.py
@@ -8,10 +8,14 @@
8
from pathlib import Path
9
torch.backends.cuda.matmul.allow_tf32 = True
10
11
+import logging
12
+
13
+vit_logger = logging.getLogger(__name__)
14
15
try:
16
from dinov3.hub.backbones import dinov3_vitl16, dinov3_vitb16
-except ImportError:
- raise ImportError("dinov3 is required for CPDINO. Install with 'pip install cellpose[dino]'.")
17
+except:
18
+ vit_logger.warning("Could not import CPDINO, run `pip install cellpose[dino]` to use CPDINO model")
19
20
21
class BaseModel(nn.Module):
0 commit comments