Fix text generation pipeline tokenizer and reorder model loading checks

app.py

@@ -22,7 +22,6 @@ _image_gen_pipeline = None
 def get_image_gen_pipeline():
     global _image_gen_pipeline
     if _image_gen_pipeline is None:
-        print("Loading image generation model on first use...")  # Optional debug message
         try:
             device = "cuda" if torch.cuda.is_available() else "cpu"
             dtype = torch.bfloat16
@@ -53,12 +52,17 @@ def get_text_gen_pipeline():
                   "mistralai/Mistral-7B-Instruct-v0.3",
                   use_fast=True  
               )
+              # Set pad_token_id to eos_token_id if pad_token is not set
+              if tokenizer.pad_token is None:
+                  tokenizer.pad_token = tokenizer.eos_token
+                  
               _text_gen_pipeline = pipeline(
                    "text-generation",
                    model="mistralai/Mistral-7B-Instruct-v0.3",
-                   tokenizer=tokenizer,  
+                   tokenizer=tokenizer,
                    max_new_tokens=2048,
                    device=device,
+                   pad_token_id=tokenizer.pad_token_id  # Explicitly set pad_token_id
               )
          except Exception as e:
               print(f"Error loading text generation model: {e}")
@@ -100,14 +104,14 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
     try:
         progress(0, desc="Starting generation...")
         
-        pipe = get_image_gen_pipeline()
-        if pipe is None:
-            return None, "Image generation model is unavailable."
-        
         # Validate that prompt is not empty
         if not prompt or prompt.strip() == "":
             return None, "Please provide a valid prompt."
 
+        pipe = get_image_gen_pipeline()
+        if pipe is None:
+            return None, "Image generation model is unavailable."
+        
         # Validate width/height dimensions
         is_valid, error_msg = validate_dimensions(width, height)
         if not is_valid: