Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix broken import in checkpoint_convert (#5635)

Browse Source 
* Fix broken import in checkpoint_convert

* simplify the fix

---------

Co-authored-by: Lincoln Stein <lstein@gmail.com>
This commit is contained in:
Lincoln Stein 2024-02-04 07:56:51 -05:00 committed by GitHub
parent 3205371654
commit 8e500283b6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/backend/model_management/convert_ckpt_to_diffusers.py
View File

@ -42,8 +42,7 @@ from diffusers.schedulers import (
PNDMScheduler, PNDMScheduler,
UnCLIPScheduler, UnCLIPScheduler,
) )
from diffusers.utils import is_accelerate_available, is_omegaconf_available from diffusers.utils import is_accelerate_available
from diffusers.utils.import_utils import BACKENDS_MAPPING
from picklescan.scanner import scan_file_path from picklescan.scanner import scan_file_path
from transformers import ( from transformers import (
AutoFeatureExtractor, AutoFeatureExtractor,
@ -1211,9 +1210,6 @@ def download_from_original_stable_diffusion_ckpt(
if prediction_type == "v-prediction": if prediction_type == "v-prediction":
prediction_type = "v_prediction" prediction_type = "v_prediction"
if not is_omegaconf_available():
raise ValueError(BACKENDS_MAPPING["omegaconf"][1])
if from_safetensors: if from_safetensors:
from safetensors.torch import load_file as safe_load from safetensors.torch import load_file as safe_load
@ -1647,11 +1643,6 @@ def download_controlnet_from_original_ckpt(
cross_attention_dim: Optional[bool] = None, cross_attention_dim: Optional[bool] = None,
scan_needed: bool = False, scan_needed: bool = False,
) -> DiffusionPipeline: ) -> DiffusionPipeline:
if not is_omegaconf_available():
raise ValueError(BACKENDS_MAPPING["omegaconf"][1])
from omegaconf import OmegaConf
if from_safetensors: if from_safetensors:
from safetensors import safe_open from safetensors import safe_open