(self, prompt, image_conditioning, max_resulting_tokens)
| 39 | ) |
| 40 | |
| 41 | def forward(self, prompt, image_conditioning, max_resulting_tokens): |
| 42 | enhanced_prompt = generate_cinematic_prompt( |
| 43 | self.image_caption_model, |
| 44 | self.image_caption_processor, |
| 45 | self.llm_model, |
| 46 | self.llm_tokenizer, |
| 47 | prompt, |
| 48 | image_conditioning, |
| 49 | max_new_tokens=max_resulting_tokens, |
| 50 | ) |
| 51 | |
| 52 | return enhanced_prompt |
| 53 | |
| 54 | @staticmethod |
| 55 | def get_model_size(model): |
nothing calls this directly
no test coverage detected