BhabhiHunter/NSFWim

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 চেক করো.")