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

doc: type hints for Generator

Browse Source
This commit is contained in:
Kevin Turner 2022-11-09 15:23:42 -08:00
parent 58ea3bf4c8
commit dcfdb83513
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ldm/invoke/generator/base.py
View File

@ -16,13 +16,19 @@ from pytorch_lightning import seed_everything
from tqdm import trange from tqdm import trange
from ldm.invoke.devices import choose_autocast from ldm.invoke.devices import choose_autocast
from ldm.models.diffusion.ddpm import DiffusionWrapper
from ldm.util import rand_perlin_2d from ldm.util import rand_perlin_2d
downsampling = 8 downsampling = 8
CAUTION_IMG = 'assets/caution.png' CAUTION_IMG = 'assets/caution.png'
class Generator(): class Generator:
def __init__(self, model, precision): downsampling_factor: int
latent_channels: int
precision: str
model: DiffusionWrapper
def __init__(self, model: DiffusionWrapper, precision: str):
self.model = model self.model = model
self.precision = precision self.precision = precision
self.seed = None self.seed = None