Commit 4ce1f330 authored by Your Name's avatar Your Name

Remove vision model alias - use image only

parent d5f7b07c
...@@ -2188,7 +2188,7 @@ class MultiModelManager: ...@@ -2188,7 +2188,7 @@ class MultiModelManager:
- "default", empty, or matches default model -> use main model - "default", empty, or matches default model -> use main model
- "audio" -> use first/default audio model - "audio" -> use first/default audio model
- "audio:modelname" -> use specific audio model - "audio:modelname" -> use specific audio model
- "vision" or "image" -> use first/default image model - "image" -> use first/default image model
- "vision:modelname" or "image:modelname" -> use specific image model - "vision:modelname" or "image:modelname" -> use specific image model
- "tts" -> use TTS model - "tts" -> use TTS model
- "tts:modelname" -> use specific TTS model - "tts:modelname" -> use specific TTS model
...@@ -2218,8 +2218,8 @@ class MultiModelManager: ...@@ -2218,8 +2218,8 @@ class MultiModelManager:
return None return None
return None return None
# Handle "vision" or "image" alias - use first/default image model # Handle "image" alias - use first/default image model
if requested_model in ("vision", "image"): if requested_model == "image":
if self.image_models: if self.image_models:
first_image = self.image_models[0] first_image = self.image_models[0]
key = f"image:{first_image}" key = f"image:{first_image}"
...@@ -2337,7 +2337,7 @@ class MultiModelManager: ...@@ -2337,7 +2337,7 @@ class MultiModelManager:
# Add vision/image models # Add vision/image models
if self.image_models: if self.image_models:
models.append(ModelInfo(id="vision")) # Alias for first image model models.append(ModelInfo(id="image")) # Alias for first image model
models.append(ModelInfo(id="image")) # Alias for first image model models.append(ModelInfo(id="image")) # Alias for first image model
# Add all image models - convert URLs to cached paths for display # Add all image models - convert URLs to cached paths for display
for image_id in self.image_models: for image_id in self.image_models:
...@@ -5007,9 +5007,9 @@ def main(): ...@@ -5007,9 +5007,9 @@ def main():
sd_model = StableDiffusion(**sd_kwargs) sd_model = StableDiffusion(**sd_kwargs)
multi_model_manager.add_model(model_key, sd_model) multi_model_manager.add_model(model_key, sd_model)
# Add aliases for "image" and "vision" # Add alias for "image"
multi_model_manager.add_model("image", sd_model) multi_model_manager.add_model("image", sd_model)
multi_model_manager.add_model("vision", sd_model)
print(f"Image model loaded successfully via sd.cpp: {original_model_name}") print(f"Image model loaded successfully via sd.cpp: {original_model_name}")
except ImportError as sd_error: except ImportError as sd_error:
print(f"stable-diffusion-cpp-python not installed: {sd_error}") print(f"stable-diffusion-cpp-python not installed: {sd_error}")
...@@ -5060,9 +5060,9 @@ def main(): ...@@ -5060,9 +5060,9 @@ def main():
pipeline = pipeline.to("cpu") pipeline = pipeline.to("cpu")
multi_model_manager.add_model(model_key, pipeline) multi_model_manager.add_model(model_key, pipeline)
# Add aliases for "image" and "vision" # Add alias for "image"
multi_model_manager.add_model("image", pipeline) multi_model_manager.add_model("image", pipeline)
multi_model_manager.add_model("vision", pipeline)
print(f"Image model loaded successfully: {model_name}") print(f"Image model loaded successfully: {model_name}")
except ImportError as e: except ImportError as e:
...@@ -5553,9 +5553,9 @@ def main(): ...@@ -5553,9 +5553,9 @@ def main():
model_key = f"image:{model_path}" model_key = f"image:{model_path}"
sd_model = StableDiffusion(**sd_kwargs) sd_model = StableDiffusion(**sd_kwargs)
multi_model_manager.add_model(model_key, sd_model) multi_model_manager.add_model(model_key, sd_model)
# Add aliases for "image" and "vision" # Add alias for "image"
multi_model_manager.add_model("image", sd_model) multi_model_manager.add_model("image", sd_model)
multi_model_manager.add_model("vision", sd_model)
print(f"Image model loaded successfully via sd.cpp: {original_model_name}") print(f"Image model loaded successfully via sd.cpp: {original_model_name}")
except ImportError as sd_error: except ImportError as sd_error:
print(f"stable-diffusion-cpp-python not installed: {sd_error}") print(f"stable-diffusion-cpp-python not installed: {sd_error}")
...@@ -5661,9 +5661,9 @@ def main(): ...@@ -5661,9 +5661,9 @@ def main():
pipeline = pipeline.to("cpu") pipeline = pipeline.to("cpu")
multi_model_manager.add_model(model_key, pipeline) multi_model_manager.add_model(model_key, pipeline)
# Add aliases for "image" and "vision" # Add alias for "image"
multi_model_manager.add_model("image", pipeline) multi_model_manager.add_model("image", pipeline)
multi_model_manager.add_model("vision", pipeline)
print(f"Image model loaded successfully: {model_name}") print(f"Image model loaded successfully: {model_name}")
except ImportError as e: except ImportError as e:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment