Akegarasu / Akegarasu/lora-scripts

Flux训练的Lora无法工作,如何正确加载?

Offen
#549 1 Kommentar 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
6.1k
Forks
699
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

```python
import torch
from optimum.quanto import freeze, qfloat8, quantize
from diffusers import FluxPipeline, FluxTransformer2DModel
from PIL import Image
from safetensors.torch import load_file

transformer = FluxTransformer2DModel.from_single_file(
"/models/Flux/FLUX.1-dev/flux1-dev-fp8.safetensors",
torch_dtype=torch.float16,
)
pipe = FluxPipeline.from_pretrained(
"/models/Flux/FLUX.1-dev",
transformer=transformer,
torch_dtype=torch.float16,
).to("cuda")

pipe.load_lora_weights("/models/output/bridge-arch-flux-000004.safetensors")

generator = torch.Generator().manual_seed(12345)
```

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.