File size: 436 Bytes
46d7d07
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from diffusers import Lumina2Transformer2DModel, Lumina2Pipeline
import torch

ckpt_path = "F:\\dos\\ComfyUI_windows_portable\\python_embeded\\v21-fix.safetensors"
transformer = Lumina2Transformer2DModel.from_single_file(
    ckpt_path, torch_dtype=torch.bfloat16
)

pipe = Lumina2Pipeline.from_pretrained(
    "Alpha-VLLM/Lumina-Image-2.0", transformer=transformer, torch_dtype=torch.bfloat16
)
pipe.save_pretrained("./v21")