Minor changes to satisfy rebase.

Author: srebrek

src/MaIN.InferPage/Components/Pages/Home.razor

@@ -386,13 +386,16 @@
         // Cloud model — pick the right generic type based on capabilities
         bool vision = Utils.Vision;
         bool reasoning = Utils.Reason;
+        bool imageGen = Utils.ImageGen;
 
         if (vision && reasoning)
             return new GenericCloudVisionReasoningModel(Id: Utils.Model, Backend: Utils.BackendType);
         if (vision)
             return new GenericCloudVisionModel(Id: Utils.Model, Backend: Utils.BackendType);
         if (reasoning)
             return new GenericCloudReasoningModel(Id: Utils.Model, Backend: Utils.BackendType);
+        if (imageGen)
+            return new GenericImageGenerationCloudModel(Id: Utils.Model, Backend: Utils.BackendType);
 
         return new GenericCloudModel(Id: Utils.Model, Backend: Utils.BackendType);
     }

src/MaIN.Services/Services/LLMService/LLMService.cs

@@ -550,11 +550,7 @@ private static BaseSamplingPipeline CreateSampler(InferenceParams interferencePa
 
     private static LocalModel GetLocalModel(Chat chat)
     {
-        // 1. Use stored model instance if available
-        if (chat.ModelInstance is LocalModel storedLocal)
-            return storedLocal;
-
-        // 2. Try registry lookup (TryGetById to avoid throwing for unregistered models)
+        // Try registry lookup (TryGetById to avoid throwing for unregistered models)
         if (ModelRegistry.TryGetById(chat.ModelId, out var model) && model is LocalModel localModel)
             return localModel;