refactor: create api module

pull/405/head
Bryce 10 months ago committed by Bryce Drennan
parent ad561e8833
commit e72e8992ab

@ -0,0 +1 @@
from .generate import imagine_image_files, imagine # noqa

@ -160,7 +160,7 @@ def imagine(
):
import torch.nn
from imaginairy.api_refiners import _generate_single_image
from imaginairy.api.generate_refiners import _generate_single_image
from imaginairy.schema import ImaginePrompt
from imaginairy.utils import (
check_torch_version,
@ -729,7 +729,7 @@ def _generate_composition_image(
):
from PIL import Image
from imaginairy.api_refiners import _generate_single_image
from imaginairy.api.generate_refiners import _generate_single_image
from imaginairy.utils import default, get_default_dtype
cutoff = normalize_image_size(cutoff)

@ -26,7 +26,7 @@ def _generate_single_image(
from refiners.foundationals.latent_diffusion.schedulers import DDIM, DPMSolver
from tqdm import tqdm
from imaginairy.api import (
from imaginairy.api.generate import (
IMAGINAIRY_SAFETY_MODE,
_generate_composition_image,
combine_image,
Loading…
Cancel
Save