(self, prompt, enable_recap, **kwargs)
| 130 | return pipeline |
| 131 | |
| 132 | def generate(self, prompt, enable_recap, **kwargs): |
| 133 | final_prompt = prompt |
| 134 | if enable_recap: |
| 135 | if not self.qwen_agent: |
| 136 | raise gr.Error("Recap is enabled, but the recap model is not available. Check model path.") |
| 137 | final_prompt = self.qwen_agent.recap_prompt(prompt) |
| 138 | |
| 139 | generator = torch.Generator(device=self.device).manual_seed(kwargs['seed']) |
| 140 | |
| 141 | with torch.inference_mode(): |
| 142 | image = self.pipeline( |
| 143 | prompt=final_prompt, |
| 144 | generator=generator, |
| 145 | num_inference_steps=kwargs['num_inference_steps'], |
| 146 | guidance_scale=kwargs['guidance_scale'], |
| 147 | width=kwargs['width'], |
| 148 | height=kwargs['height'] |
| 149 | ).images[0] |
| 150 | |
| 151 | return image, final_prompt |
| 152 | |
| 153 | # --- Global Model Initialization --- |
| 154 | generator = PosterGenerator( |
no test coverage detected