diff --git a/packages/inference/src/snippets/templates/python/fal_client/textToImage.jinja b/packages/inference/src/snippets/templates/python/fal_client/textToImage.jinja index 869f5fb19e..6dc58769a2 100644 --- a/packages/inference/src/snippets/templates/python/fal_client/textToImage.jinja +++ b/packages/inference/src/snippets/templates/python/fal_client/textToImage.jinja @@ -1,11 +1,21 @@ {% if provider == "fal-ai" %} import fal_client +{% if providerInputs.asObj.loras is defined and providerInputs.asObj.loras != none %} result = fal_client.subscribe( "{{ providerModelId }}", arguments={ "prompt": {{ inputs.asObj.inputs }}, + "loras":{{ providerInputs.asObj.loras | tojson }}, }, ) +{% else %} +result = fal_client.subscribe( + "{{ providerModelId }}", + arguments={ + "prompt": {{ inputs.asObj.inputs }}, + }, +) +{% endif %} print(result) {% endif %} \ No newline at end of file diff --git a/packages/tasks-gen/snippets-fixtures/text-to-image--lora/python/fal_client/0.fal-ai.py b/packages/tasks-gen/snippets-fixtures/text-to-image--lora/python/fal_client/0.fal-ai.py index a0eb846623..61ed08a3f9 100644 --- a/packages/tasks-gen/snippets-fixtures/text-to-image--lora/python/fal_client/0.fal-ai.py +++ b/packages/tasks-gen/snippets-fixtures/text-to-image--lora/python/fal_client/0.fal-ai.py @@ -4,6 +4,8 @@ "", arguments={ "prompt": "Astronaut riding a horse", + "loras":[{"path": "https://huggingface.co/openfree/flux-chatgpt-ghibli-lora/resolve/main/", "scale": 1}], }, ) + print(result) \ No newline at end of file diff --git a/packages/tasks-gen/snippets-fixtures/text-to-image/python/fal_client/0.fal-ai.py b/packages/tasks-gen/snippets-fixtures/text-to-image/python/fal_client/0.fal-ai.py index 461dbb2271..87d2649343 100644 --- a/packages/tasks-gen/snippets-fixtures/text-to-image/python/fal_client/0.fal-ai.py +++ b/packages/tasks-gen/snippets-fixtures/text-to-image/python/fal_client/0.fal-ai.py @@ -6,4 +6,5 @@ "prompt": "Astronaut riding a horse", }, ) + print(result) \ No newline at end of file