Diffusers documentation

توليد الصور من العمق بتوجيه النص

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

توليد الصور من العمق بتوجيه النص

يتيح لك StableDiffusionDepth2ImgPipeline تمرير موجه نصي وصورة أولية لتوجيه توليد صور جديدة. بالإضافة إلى ذلك، يمكنك أيضًا تمرير depth_map للحفاظ على بنية الصورة. إذا لم يتم توفير depth_map، فإن خط الأنابيب يتنبأ تلقائيًا بالعمق عبر نموذج تقدير العمق المدمج.

ابدأ بإنشاء كائن من StableDiffusionDepth2ImgPipeline:

import torch
from diffusers import StableDiffusionDepth2ImgPipeline
from diffusers.utils import load_image, make_image_grid

pipeline = StableDiffusionDepth2ImgPipeline.from_pretrained(
    "stabilityai/stable-diffusion-2-depth",
    dtype=torch.float16,
    use_safetensors=True,
).to("cuda")  # or "mps", "xpu", "cpu"

الآن مرّر موجهك إلى خطّ الأنابيب. يمكنك أيضًا تمرير negative_prompt لمنع كلمات معينة من توجيه كيفية توليد الصورة:

url = "http://images.cocodataset.org/val2017/000000039769.jpg"
init_image = load_image(url)
prompt = "two tigers"
negative_prompt = "bad, deformed, ugly, bad anatomy"
image = pipeline(prompt=prompt, image=init_image, negative_prompt=negative_prompt, strength=0.7).images[0]
make_image_grid([init_image, image], rows=1, cols=2)
المدخلالمخرجات
Update on GitHub