diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py
index af26d0a985..77f2430454 100644
--- a/invokeai/backend/model_manager/probe.py
+++ b/invokeai/backend/model_manager/probe.py
@@ -132,7 +132,7 @@ class ModelProbe(object):
 
         format_type = ModelFormat.Diffusers if model_path.is_dir() else ModelFormat.Checkpoint
         model_info = None
-        model_type = fields['type'] if 'type' in fields else None
+        model_type = ModelType(fields['type']) if 'type' in fields else None
         if not model_type:
             if format_type is ModelFormat.Diffusers:
                 model_type = cls.get_model_type_from_folder(model_path)