Skip to content

Commit e350527

Browse files
authored
[inference] replicate: add support for i2i LoRA (#1656)
Ctx (private): https://huggingface.slack.com/archives/C050PS98HBM/p1753381564315359
1 parent 34c67e2 commit e350527

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

packages/inference/src/providers/replicate.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,10 @@ export class ReplicateImageToImageTask extends ReplicateTask implements ImageToI
170170
...omit(params.args, ["inputs", "parameters"]),
171171
...params.args.parameters,
172172
input_image: params.args.inputs, // This will be processed in preparePayloadAsync
173+
lora_weights:
174+
params.mapping?.adapter === "lora" && params.mapping.adapterWeightsPath
175+
? `https://huggingface.co/${params.mapping.hfModelId}`
176+
: undefined,
173177
},
174178
version: params.model.includes(":") ? params.model.split(":")[1] : undefined,
175179
};

0 commit comments

Comments
 (0)