Views
No views yet
A photo-realistic image of a cat4.00.020None421024x10241{
2 "algo": "lokr",
3 "multiplier": 1.0,
4 "linear_dim": 10000,
5 "linear_alpha": 1,
6 "factor": 4,
7 "apply_preset": {
8 "target_module": [
9 "FluxTransformer2DModel"
10 ],
11 "module_algo_map": {
12 "Attention": {
13 "factor": 4
14 },
15 "FeedForward": {
16 "factor": 2
17 }
18 }
19 }
20}1import torch
2from diffusers import DiffusionPipeline
3from lycoris import create_lycoris_from_weights
4
5model_id = 'terminusresearch/flux-booru-v0.2'
6adapter_id = 'pytorch_lora_weights.safetensors' # you will have to download this manually
7lora_scale = 1.0
8wrapper, _ = create_lycoris_from_weights(lora_scale, adapter_id, pipeline.transformer)
9wrapper.merge_to()
10
11prompt = "A photo-realistic image of a cat"
12
13pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
14image = pipeline(
15 prompt=prompt,
16 num_inference_steps=20,
17 generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
18 width=1024,
19 height=1024,
20 guidance_scale=4.0,
21).images[0]
22image.save("output.png", format="PNG")