diff --git a/api/service/generation_service.py b/api/service/generation_service.py index 772f0c6..819bc53 100644 --- a/api/service/generation_service.py +++ b/api/service/generation_service.py @@ -209,6 +209,7 @@ class GenerationService: result_list=[new_asset.id], result=new_asset.id, progress=100, + nsfw=external_gen.nsfw, execution_time_seconds=external_gen.execution_time_seconds, api_execution_time_seconds=external_gen.api_execution_time_seconds, token_usage=external_gen.token_usage,