@@ -51,6 +51,14 @@ class Fluxlorafast(SimpleService):
         return messages
 
     def make(self, input_message: Message, save: bool = True) -> list[Message]:
+        sizes = {
+            '1:1': 'square',
+            '1:1 HD': 'square_hd',
+            'Портрет 4:3': 'portrait_4_3',
+            'Портрет 16:9': 'portrait_16_9',
+            '4:3': 'landscape_4_3',
+            '16:9': 'landscape_16_9',
+        }
         start_time = time.time()
         version = input_message.info.get('version')
         translated_prompt = self.translate_prompt(input_message.content)
@@ -58,6 +66,7 @@ class Fluxlorafast(SimpleService):
             {
                 'prompt': translated_prompt,
                 'model_version': 'fb90c17a-d410-41e7-9961-dc7c687bc627',
+                'image_size': sizes.get(input_message.info.pop('image_size'), '4:3'),
                 **input_message.info,
             }
         )