Diffusers documentation
AutoModel
AutoModel
AutoModel, kontrol noktası config.json dosyasından doğru model sınıfını otomatik olarak alır.
AutoModel
from_pretrained
< source >( pretrained_model_or_path: str | os.PathLike | None = None**kwargs )
Parametreler
- pretrained_model_name_or_path (
strveyaos.PathLike, isteğe bağlı) — Şu seçeneklerden biri olabilir:- Bir dize, Hub üzerinde barındırılan önceden eğitilmiş bir modelin model kimliği (örneğin
google/ddpm-celebahq-256). - Model ağırlıklarının save_pretrained() ile kaydedildiği bir dizin (örneğin
./my_model_directory) yolunu belirtir.
- Bir dize, Hub üzerinde barındırılan önceden eğitilmiş bir modelin model kimliği (örneğin
- cache_dir (
str | os.PathLike, isteğe bağlı) — Standart önbellek kullanılmıyorsa, indirilen önceden eğitilmiş model yapılandırmasının önbelleğe alındığı dizinin yolu. - dtype (
torch.dtype, isteğe bağlı) — Varsayılantorch.dtypedeğerini geçersiz kılar ve modeli başka bir dtype ile yükler. - force_download (
bool, isteğe bağlı, varsayılanFalse) — Model ağırlıklarının ve yapılandırma dosyalarının (yeniden) indirilmesinin zorlanıp zorlanmayacağını belirler; önbellekteki sürümler varsa bunların üzerine yazar. - proxies (
dict[str, str], isteğe bağlı) — Protokole veya uç noktaya göre kullanılacak vekil sunucuların sözlüğü, örneğin{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}. Vekiller her istekte kullanılır. - output_loading_info (
bool, isteğe bağlı, varsayılanFalse) — Eksik anahtarlar, beklenmeyen anahtarlar ve hata mesajlarını içeren bir sözlüğün de döndürülüp döndürülmeyeceğini belirtir. - local_files_only(
bool, isteğe bağlı, varsayılanFalse) — Yalnızca yerel model ağırlıkları ve yapılandırma dosyalarının yüklenip yüklenmeyeceğini belirtir.Trueolarak ayarlanırsa, model Hub'dan indirilmez. - token (
strveya bool, isteğe bağlı) — Uzak dosyalar için HTTP bearer yetkilendirmesi olarak kullanılacak jeton.Trueise,diffusers-cli loginile oluşturulan jeton (~/.huggingfaceiçinde saklanır) kullanılır. - revision (
str, isteğe bağlı, varsayılan"main") — Kullanılacak belirli model sürümü. Bir dal adı, etiket adı, commit kimliği veya Git tarafından izin verilen herhangi bir tanımlayıcı olabilir. - subfolder (
str, isteğe bağlı, varsayılan"") — Hub'da veya yerel olarak daha büyük bir model deposu içindeki bir model dosyasının alt klasör konumu. - mirror (
str, isteğe bağlı) — Çin'de bir model indirirken erişilebilirlik sorunlarını çözmek için kullanılacak ayna kaynağı. Kaynağın güncelliğini veya güvenliğini garanti etmiyoruz; daha fazla bilgi için lütfen ayna sitesine başvurun. - device_map (
strveyadict[str, int | str | torch.device], isteğe bağlı) — Her alt modülün nereye yerleştirileceğini belirten bir eşleme. Her parametre/buffer adı için tanımlanması gerekmez; belirli bir modül adı bir kez dahil edildiğinde, onun tüm alt modülleri aynı cihaza gönderilir. Varsayılan değerNone, anlamına gelir, yani model CPU üzerinde yüklenecektir.🤗 Accelerate'in en optimize
device_map'i otomatik olarak hesaplaması içindevice_map="auto"ayarlayın. Her seçenek hakkında daha fazla bilgi için cihaz haritası tasarımı bölümüne bakın. - max_memory (
Dict, isteğe bağlı) — Maksimum bellek için bir sözlük cihaz tanımlayıcısı. Ayarlanmazsa, varsayılan olarak her GPU için kullanılabilir maksimum bellek ve kullanılabilir CPU RAM'i olur. - offload_folder (
strveyaos.PathLike, isteğe bağlı) —device_mapdeğeri"disk"içeriyorsa ağırlıkların taşınacağı yol. - offload_state_dict (
bool, isteğe bağlı) —Trueise, CPU state dict + checkpoint'in en büyük shard'ı sığmıyorsa CPU RAM'inin dolmasını önlemek için CPU state dict'i geçici olarak sabit diske taşır. Disk offload varsa varsayılan olarakTrueolur. - low_cpu_mem_usage (
bool, isteğe bağlı, varsayılan olarak torch sürümü >= 1.9.0 iseTrue, aksi haldeFalse) — Modeli yalnızca önceden eğitilmiş ağırlıkları yükleyerek ve ağırlıkları başlatmadan yüklemeyi hızlandırır. Bu, model yüklenirken (tepe bellek dahil) CPU belleğinde model boyutunun 1 katından fazlasını kullanmamaya da çalışır. Yalnızca PyTorch >= 1.9.0 için desteklenir. PyTorch'un daha eski bir sürümünü kullanıyorsanız, bu argümanıTrueolarak ayarlamak bir hata verecektir. - variant (
str, isteğe bağlı) — Ağırlıkları,"fp16"veya"ema"gibi belirli birvariantdosya adından yükler. - use_safetensors (
bool, isteğe bağlı, varsayılanNone) —Noneolarak ayarlanırsa,safetensorsağırlıkları mevcut vesafetensorskütüphanesi yüklüyse indirilir.Trueolarak ayarlanırsa, modelsafetensorsağırlıklarından zorla yüklenir.Falseolarak ayarlanırsa,safetensorsağırlıkları yüklenmez. - disable_mmap (‘bool’, isteğe bağlı, varsayılan ‘False’) — Bir Safetensors modeli yüklenirken mmap'in devre dışı bırakılıp bırakılmayacağı. Model bir ağ biriminde veya sabit diskte olduğunda, bu ortamın mmap'in sıklıkla konum değiştirme (seek) gerektiren doğasını iyi yönetememesi nedeniyle bu seçenek daha iyi performans gösterebilir.
- trust_remote_cocde (
bool, isteğe bağlı, varsayılanFalse) — Uzak kodun güvenilip güvenilmeyeceği
Önceden eğitilmiş bir model yapılandırmasından önceden eğitilmiş bir PyTorch modeli örnekleyin.
Model varsayılan olarak değerlendirme moduna ayarlanır - model.eval() - ve dropout modülleri devre dışı bırakılır. Modeli eğitmek için model.train() ile eğitim moduna geri ayarlayın.
> Özel veya kısıtlı modelleri kullanmak için
hf > auth loginile oturum açın. Ayrıca, bu yöntemi duvarlı bir ortamda kullanmak için özel > “çevrimdışı mod”u etkinleştirebilirsiniz.
Örnek:
from diffusers import AutoModel
unet = AutoModel.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="unet")Aşağıdaki hata mesajını alıyorsanız, aşağı akış göreviniz için ağırlıkları ince ayar yapmanız gerekir:
Some weights of UNet2DConditionModel were not initialized from the model checkpoint at stable-diffusion-v1-5/stable-diffusion-v1-5 and are newly initialized because the shapes did not match:
- conv_in.weight: found shape torch.Size([320, 4, 3, 3]) in the checkpoint and torch.Size([320, 9, 3, 3]) in the model instantiated
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.from_config
< source >( pretrained_model_name_or_path_or_dict: str | os.PathLike | dict | None = None**kwargs )
Parametreler
- pretrained_model_name_or_path_or_dict (
str,os.PathLike, veyadict) — Şunlardan biri olabilir:- Hub'da barındırılan önceden eğitilmiş bir modelin model kimliği (örneğin
google/ddpm-celebahq-256) olan bir dize - Model yapılandırma dosyası içeren bir dizin (örneğin
./my_model_directory) yoluna. - Bir yapılandırma sözlüğü.
- Hub'da barındırılan önceden eğitilmiş bir modelin model kimliği (örneğin
- cache_dir (
Union[str, os.PathLike], isteğe bağlı) — Standart önbellek kullanılmıyorsa, indirilen önceden eğitilmiş model yapılandırmasının önbelleğe alındığı dizinin yolu. - force_download (
bool, isteğe bağlı, varsayılanFalse) — Model yapılandırmasının (yeniden) indirilmesinin zorlanıp zorlanmayacağı; önbellekte bir sürüm varsa bu sürümün geçersiz kılınması. - proxies (
Dict[str, str], isteğe bağlı) — Protokole veya uç noktaya göre kullanılacak vekil sunucuların sözlüğü. - local_files_only(
bool, isteğe bağlı, varsayılanFalse) — Yalnızca yerel model yapılandırma dosyalarının yüklenip yüklenmeyeceğini belirtir. - token (
strveya bool, isteğe bağlı) — Uzak dosyalar için HTTP bearer yetkilendirmesi olarak kullanılacak jeton. - revision (
str, isteğe bağlı, varsayılan"main") — Kullanılacak belirli model sürümü. - trust_remote_code (
bool, isteğe bağlı, varsayılanFalse) — Uzak kodun güvenilip güvenilmeyeceğini belirtir. - subfolder (
str, isteğe bağlı, varsayılan"") — Hub'da veya yerel olarak daha büyük bir model deposu içindeki bir model dosyasının alt klasör konumu.
Rastgele ağırlıklarla bir yapılandırma sözlüğünden veya önceden eğitilmiş bir model yapılandırma dosyasından model örnekleyin (önceden eğitilmiş ağırlıklar yüklenmez).