Views
No views yet
Generate retro RPG game sprite pixel art with AI! / Buat pixel art bergaya RPG klasik dengan AI!


1!pip install diffusers transformers accelerate peft -q
2!pip install torchao --upgrade -q
3
4import torch
5from diffusers import StableDiffusionPipeline
6from peft import PeftModel
7from IPython.display import display
8
9pipe = StableDiffusionPipeline.from_pretrained(
10 "stable-diffusion-v1-5/stable-diffusion-v1-5",
11 torch_dtype=torch.float16,
12 safety_checker=None,
13).to("cuda")
14
15pipe.unet = PeftModel.from_pretrained(pipe.unet, "VaroDZAKY/Varo_pixel_Art")
16pipe.unet = pipe.unet.to("cuda")
17
18def generate(prompt):
19 pipe.unet.eval()
20 result = pipe(
21 prompt=f"pixelart_style, {prompt}, sharp pixel edges, flat colors, no gradient",
22 negative_prompt="blurry, realistic, 3d, gradient, smooth shading, watermark, text",
23 num_inference_steps=100,
24 guidance_scale=11.0,
25 height=512,
26 width=512,
27 ).images[0]
28 display(result)
29 return result
30
31# Just type the object name! / Tinggal ketik nama objek!
32generate("monster with sword")
33generate("red fire dragon")
34generate("ice monster")