Skip to content

update doc string #1469 and import error message #1470

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
carsen-stringer merged 1 commit into from
Jun 11, 2026
+4 −4
Merged

update doc string #1469 and import error message #1470

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions cellpose/models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
try:
from .vit import CPDINO
except:
models_logger.warning("Could not import CPDINO, run `pip install cellpose[dino]` to use CPDINO model")
models_logger.warning("Could not import CPDINO, run `pip install git+https://github.com/facebookresearch/dinov3` to use CPDINO model")
from .core import assign_device, run_net, run_3D

_MODEL_URL = "https://huggingface.co/mouseland/cellpose-sam/resolve/main/"
Expand Down Expand Up @@ -108,9 +108,9 @@ def __init__(self, gpu=False, pretrained_model="cpsam_v2", model_type=None,

Parameters:
gpu (bool, optional): Whether or not to save model to GPU, will check if GPU available.
pretrained_model (str or list of strings, optional): Full path to pretrained cellpose model(s), if None or False, no model loaded.
pretrained_model (str): Name of pretrained model ("cpdino", "cpsam_v2", etc) or path to pretrained cellpose model. Default is "cpsam_v2".
model_type (str, optional): Any model that is available in the GUI, use name in GUI e.g. "livecell" (can be user-trained or model zoo).
diam_mean (float, optional): Mean "diameter", 30. is built-in value for "cyto" model; 17. is built-in value for "nuclei" model; if saved in custom model file (cellpose>=2.0) then it will be loaded automatically and overwrite this value.
diam_mean (float, optional): Deprecated in v4.0.1+, not used.
device (torch device, optional): Device used for model running / training (torch.device("cuda") or torch.device("cpu")), overrides gpu input, recommended if you want to use a specific GPU (e.g. torch.device("cuda:1")).
use_bfloat16 (bool, optional): Use 16bit float precision instead of 32bit for model weights. Default to 16bit (True).
"""
Expand Down
2 changes: 1 addition & 1 deletion cellpose/vit.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
try:
from dinov3.hub.backbones import dinov3_vitl16, dinov3_vitb16
except:
vit_logger.warning("Could not import CPDINO, run `pip install cellpose[dino]` to use CPDINO model")
vit_logger.warning("Could not import CPDINO, run `pip install git+https://github.com/facebookresearch/dinov3` to use CPDINO model")


class BaseModel(nn.Module):
Expand Down
Loading