Skip to content

Plans on LyCORIS support? #52

Open
@JakobLS

Description

@JakobLS

Do you have any plans on supporting LyCORIS weights as well, a part from current LoRA?

When I try to apply them on the INT4 pipeline as follows

transformer = NunchakuFluxTransformer2dModel.from_pretrained("mit-han-lab/svdq-int4-flux.1-dev")
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    transformer=transformer,
    torch_dtype=torch.bfloat16
)

# Apply LoCORIS weights
lycoris_net, _ = create_lycoris_from_weights(1, weight_path, t2i_pipe.transformer)
lycoris_net.apply_to()

the code completes without erroring out:

2024-12-11 16:41:20|[LyCORIS]-INFO: Loading Modules from state dict...
2024-12-11 16:41:20|[LyCORIS]-INFO: 0 Modules Loaded

But 0 modules are loaded, and I get no effect at all on the resulting images.

When I apply the weights on the bf16 model, 342 modules are loaded and I do get an effect. It looks like the modules are not properly loaded onto the pipeline.

Or perhaps this is more on LYCORIS's side?

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions