Akegarasu / Akegarasu/lora-scripts
Flux训练的Lora无法工作，如何正确加载？
- Ngôn ngữ chính
- Python
- Star
- 6.1k
- Fork
- 699
- Chỉ số merge pull request
- Không có pull request nào được merge trong 30 ngày
Mô tả
```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)
```
Hướng dẫn đóng góp
Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này
Đánh giá
Issue này chưa được đánh giá.