@@ -77,8 +77,8 @@ 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'),
-                'loras': [loras.get(input_message.info.pop('loras'), 'Realistic')],
+                'image_size': sizes.get(input_message.info.pop('image_size', None), '4:3'),
+                'loras': [loras.get(input_message.info.pop('loras', None), 'Realistic')],
                 **input_message.info,
             }
         )