diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index ee89764..b387a07 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.138' +__version__ = '8.0.139' from ultralytics.engine.model import YOLO from ultralytics.hub import start diff --git a/ultralytics/cfg/__init__.py b/ultralytics/cfg/__init__.py index 75703de..65aae08 100644 --- a/ultralytics/cfg/__init__.py +++ b/ultralytics/cfg/__init__.py @@ -372,6 +372,9 @@ def entrypoint(debug=''): if 'rtdetr' in model.lower(): # guess architecture from ultralytics import RTDETR model = RTDETR(model) # no task argument + elif 'fastsam' in model.lower(): + from ultralytics import FastSAM + model = FastSAM(model) elif 'sam' in model.lower(): from ultralytics import SAM model = SAM(model)