Views
No views yet
1import torch
2from diffusers import AutoPipelineForImage2Image
3from diffusers.utils import load_image
4
5# Load the base Turbo model
6pipe = AutoPipelineForImage2Image.from_pretrained(
7 "Tongyi-MAI/Z-Image-Turbo",
8 torch_dtype=torch.float16,
9 variant="fp16"
10).to("cuda")
11
12# Load this Cyberpunk LoRA
13pipe.load_lora_weights("srmahapatra95/Cyberpunk-Lora-Weights")
14
15# Initial Image
16init_image = load_image("[https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img-init.png](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img-init.png)")
17
18# Prompt
19prompt = "cyberpunk style, futuristic city, neon lights, high tech, rain, 8k"
20
21# Generate (Keep steps low for Turbo models)
22image = pipe(prompt, image=init_image, strength=0.6, num_inference_steps=4).images[0]
23image.save("cyberpunk_output.png")