| 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") |