Diffusers documentation

Intel Gaudi

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Intel Gaudi

تشمل عائلة معالجات Intel Gaudi للذكاء الاصطناعي Intel Gaudi 1 وIntel Gaudi 2 وIntel Gaudi 3. كل خادم مزوّد بثمانية أجهزة تُعرف بوحدات معالجة Habana (HPU)، وتوفر 128 جيجابايت من الذاكرة في Gaudi 3، و96 جيجابايت في Gaudi 2، و32 جيجابايت في الجيل الأول من Gaudi. لمزيد من التفاصيل حول بنية العتاد الأساسية، راجع نظرة عامة على Gaudi Architecture.

يمكن أن تستفيد خطوط أنابيب Diffusers من تسريع HPU، حتى لو لم يُضف خط أنابيب معين إلى Optimum for Intel Gaudi بعد، وذلك باستخدام GPU Migration Toolkit.

استدعِ .to("hpu") على خطّ الأنابيب الخاص بك لنقله إلى جهاز HPU كما هو موضح أدناه بالنسبة إلى Flux:

import torch
from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", dtype=torch.bfloat16)
pipeline.to("hpu")

image = pipeline("An image of a squirrel in Picasso style").images[0]

لتنفيذ خطوط أنابيب الانتشار المُحسّنة لـ Gaudi، نوصي باستخدام Optimum for Intel Gaudi.

Update on GitHub