pip install torch diffusers transformers accelerate
import torch
from diffusers import StableDiffusionPipeline
মডেল লোড করো (uncensored ভার্সনের জন্য safety checker অফ করো)
pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4",
torch_dtype=torch.float16 # GPU-তে ফাস্ট করার জন্য
)
pipe.safety_checker = lambda images, clip_input: (images, False) # Safety checker ডিসেবল করে uncensored করো
pipe = pipe.to("cuda") # যদি GPU থাকে, না হলে "cpu" দাও
প্রম্পট দিয়ে ইমেজ জেনারেট করো
prompt = "a futuristic city at night" # তোমার প্রম্পট এখানে চেঞ্জ করো
image = pipe(prompt).images[0]
ইমেজ সেভ করো
image.save("generated_image.png")
print("ইমেজ জেনারেট হয়েছে! generated_image.png চেক করো.")
