@@ -22,10 +22,11 @@ class Stablediffusion(SimpleService): contains abstract method make, which makes a generation """ - MODELS = ['sd3', 'sd3-turbo'] + MODELS = ['sd3', 'sd3-turbo', 'sd3-medium'] MODELS_LINKS = { 'sd3': 'stable-diffusion-3.5-large', 'sd3-turbo': 'stable-diffusion-3.5-large-turbo', + 'sd3-medium': 'stable-diffusion-3.5-medium' } def calculate_price(self, input_message: Message) -> Decimal: @@ -33,6 +34,8 @@ class Stablediffusion(SimpleService): return Decimal('13') elif input_message.info.get('version') == 'sd3-turbo': return Decimal('8') + elif input_message.info.get('version') == 'sd3-medium': + return Decimal('7') def save_results( self,