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

generate: fix import error

Browse Source
This commit is contained in:
Kevin Turner 2022-12-12 16:17:46 -08:00
parent 0b48f2e801
commit 31fd285c82
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ldm/generate.py
View File

@ -16,6 +16,7 @@ import numpy as np
import skimage
import torch
import transformers
from PIL import Image, ImageOps
from diffusers import HeunDiscreteScheduler
from diffusers.pipeline_utils import DiffusionPipeline
from diffusers.schedulers.scheduling_ddim import DDIMScheduler
@ -26,9 +27,9 @@ from diffusers.schedulers.scheduling_ipndm import IPNDMScheduler
from diffusers.schedulers.scheduling_lms_discrete import LMSDiscreteScheduler
from diffusers.schedulers.scheduling_pndm import PNDMScheduler
from omegaconf import OmegaConf
from PIL import Image, ImageOps
from pytorch_lightning import seed_everything, logging
import ldm.invoke.conditioning
from ldm.invoke.args import metadata_from_png
from ldm.invoke.concepts_lib import Concepts
from ldm.invoke.conditioning import get_uc_and_c_and_ec
@ -616,9 +617,9 @@ class Generate:
# used by multiple postfixers
# todo: cross-attention control
uc, c, extra_conditioning_info = get_uc_and_c_and_ec(
prompt, model =self.model,
prompt, model=self.model,
skip_normalize_legacy_blend=opt.skip_normalize,
log_tokens =ldm.invoke.conditioning.log_tokenization
log_tokens=ldm.invoke.conditioning.log_tokenization
)
if tool in ('gfpgan','codeformer','upscale'):