DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-01-08 11:57:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add AutoencoderKL to the list of models that opt-out of partial loading.

Browse Source
This commit is contained in:
Ryan Dick 2024-12-19 15:25:23 -05:00
parent f9e899a6ba
commit 609ed06265
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/load/model_cache/model_cache.py
View File

@ -6,6 +6,7 @@ from typing import Dict, List, Optional
import psutil
import torch
from diffusers.models.autoencoders.autoencoder_kl import AutoencoderKL
from invokeai.backend.flux.ip_adapter.xlabs_ip_adapter_flux import XlabsIpAdapterFlux
from invokeai.backend.flux.modules.autoencoder import AutoEncoder
@ -156,6 +157,7 @@ class ModelCache:
AutoEncoder,
# XLabsIPAdapterFlux is a wrapper around two models that are called directly.
XlabsIpAdapterFlux,
AutoencoderKL,
)
# Wrap model.