@@ -31,7 +31,8 @@ class Flux(SimpleService): description = 'Нейросеть, способная генерировать картинки из вашего текста' category = ModelCategory(title='Изображения', slug='images') versions = [ - ModelVersion(name='Flux-Pro1.1', slug='flux-pro-1.1', default=True), + ModelVersion(name='Flux-Schnell', slug='flux-schnell', default=True), + ModelVersion(name='Flux-Pro1.1', slug='flux-pro-1.1'), ModelVersion(name='Flux-Dev', slug='flux-dev'), ModelVersion(name='Ultra', slug='flux-1.1-pro-ultra'), ] @@ -88,6 +89,12 @@ class Flux(SimpleService): type=ModelParameter.TypeChoices.INTRANGE, values={'start': 1, 'end': 50, 'step': 1, 'default': 28}, ), + ModelParameter( + name='Количество шагов вывода', + key='num_inference_steps', + type=ModelParameter.TypeChoices.INTRANGE, + values={'start': 1, 'end': 4, 'step': 1, 'default': 4}, + ), ModelParameter( name='Соотношение сторон', key='aspect_ratio', @@ -132,16 +139,22 @@ class Flux(SimpleService): versions[0].slug: ModelPaymentRule( strategy=ModelPaymentRule.StrategyChoices.FIXED, interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT, - cost=4.4, + cost=0.33, coefficient=5.00, ), versions[1].slug: ModelPaymentRule( strategy=ModelPaymentRule.StrategyChoices.FIXED, interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT, - cost=2.75, + cost=4.4, coefficient=5.00, ), versions[2].slug: ModelPaymentRule( + strategy=ModelPaymentRule.StrategyChoices.FIXED, + interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT, + cost=2.75, + coefficient=5.00, + ), + versions[3].slug: ModelPaymentRule( strategy=ModelPaymentRule.StrategyChoices.FIXED, interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT, cost=6.6, @@ -17,9 +17,10 @@ services: python manage.py compilemessages python -m uvicorn --host 0.0.0.0 --workers 8 --timeout-keep-alive 300 --ws wsproto --loop uvloop --http httptools --lifespan off --log-level info backend.asgi:application networks: + - default - infrastructure deploy: - replicas: 2 + replicas: 1 update_config: parallelism: 1 delay: 10s @@ -71,6 +72,8 @@ services: context: . dockerfile: Dockerfile command: celery -A backend worker -l INFO --concurrency 8 + networks: + - default deploy: replicas: 1 update_config: @@ -96,6 +99,8 @@ services: context: . dockerfile: Dockerfile command: celery -A backend beat -l INFO + networks: + - default deploy: replicas: 1 update_config: @@ -156,6 +161,8 @@ services: celery-mdb: image: redis:alpine + networks: + - default deploy: replicas: 1 restart_policy: