Add prod

prod_model.py

@@ -33,7 +33,9 @@ class ProductionDesignModel(DesignModel):
             logging.info(f"Using device: {self.device}")
             
             self.model_id = "stabilityai/stable-diffusion-2-1"
+            self.tokenizer_id = "openai/clip-vit-large-patch14"  # Correct tokenizer for SD 2.1
             logging.info(f"Loading model: {self.model_id}")
+            logging.info(f"Loading tokenizer: {self.tokenizer_id}")
             
             # Initialize the pipeline with error handling
             try:
@@ -55,8 +57,13 @@ class ProductionDesignModel(DesignModel):
                 logging.error(f"Error loading model: {e}")
                 raise
             
-            # Initialize tokenizer
-            self.tokenizer = CLIPTokenizer.from_pretrained(self.model_id)
+            # Initialize tokenizer with correct path
+            try:
+                self.tokenizer = CLIPTokenizer.from_pretrained(self.tokenizer_id)
+                logging.info("Tokenizer loaded successfully")
+            except Exception as e:
+                logging.error(f"Error loading tokenizer: {e}")
+                raise
             
             # Set default prompts
             self.neg_prompt = "blurry, low quality, distorted, deformed, disfigured, watermark, text, bad proportions, duplicate, double, multiple, broken, cropped"