diff --git a/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py b/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py index d613879..54c9ed5 100644 --- a/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py +++ b/nostr_dvm/tasks/imagegeneration_replicate_fluxpro.py @@ -66,8 +66,8 @@ class ImageGenerationReplicateFluxPro(DVMTaskInterface): async def create_request_from_nostr_event(self, event, client=None, dvm_config=None): request_form = {"jobID": event.id().to_hex() + "_" + self.NAME.replace(" ", "")} prompt = "" - width = "1024" - height = "1024" + width = "4" + height = "5" for tag in event.tags().to_vec(): @@ -94,7 +94,7 @@ class ImageGenerationReplicateFluxPro(DVMTaskInterface): options = { "prompt": prompt, - "size": width + "x" + height, + "ratio": width + ":" + height, "number": 1 } request_form['options'] = json.dumps(options) @@ -111,7 +111,7 @@ class ImageGenerationReplicateFluxPro(DVMTaskInterface): output = replicate.run( "black-forest-labs/flux-1.1-pro", input={"prompt": options["prompt"], - "aspect_ratio": "4:5", + "aspect_ratio": options["ratio"], "output_format": "jpg", "output_quality": 80, "safety_tolerance": 2, diff --git a/tests/otherstuff.py b/tests/otherstuff.py index 6c7f12b..265cef7 100644 --- a/tests/otherstuff.py +++ b/tests/otherstuff.py @@ -315,7 +315,7 @@ def build_replicate(name, identifier, model, announce): nip89info = { "name": name, "picture": "https://i.nostr.build/qnoBIN4jSkfF8IHk.png", - "about": "I use Replicate to run StableDiffusion XL", + "about": "I make images with StableDiffusion XL", "supportsEncryption": True, "acceptsNutZaps": False, "nip90Params": { @@ -352,7 +352,7 @@ def build_replicate_recraft(name, identifier, announce): nip89info = { "name": name, "picture": "https://i.nostr.build/jSbrXvYglXCzSeAc.jpg", - "about": "I use Replicate to run Recraft v3", + "about": "I produce images with Recraft v3", "supportsEncryption": True, "acceptsNutZaps": False, "nip90Params": { @@ -387,13 +387,13 @@ def build_replicate_fluxpro(name, identifier, announce): nip89info = { "name": name, "picture": "https://i.nostr.build/AQTujqzVmLxLmG16.jpg", - "about": "I use Replicate to FluxPro 1.1.", + "about": "I provide images with FluxPro 1.1", "supportsEncryption": True, "acceptsNutZaps": False, "nip90Params": { "size": { "required": False, - "values": ["5:4"] + "values": ["4:5"] } } } @@ -442,6 +442,8 @@ def playground(announce=False): bot_config.SUPPORTED_DVMS.append(recraftreplicate) recraftreplicate.run() + + media_bringer = build_media_converter("Nostr AI DVM Media Converter", "media_converter", announce) #bot_config.SUPPORTED_DVMS.append(media_bringer)