aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-01-24 16:04:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Pre release changes for production (#59)

Browse Source 
* clean requirements

* rm taming deps

* isort, black

* mv lipips, license

* clean vq, fix path

* fix loss path, gitignore

* tested requirements pt13

* fix numpy req for python3.8, add tests

* fix name

* fix dep scipy 3.8 pt2

* add black test formatter
This commit is contained in:
Benjamin Aubin
2023-07-26 12:09:28 +02:00
committed by GitHub
parent 4a3f0f546e
commit e596332148
31 changed files with 642 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgm/modules/diffusionmodules/__init__.py
View File

@@ -1,7 +1,7 @@
from .denoiser import Denoiser
from .discretizer import Discretization
from .loss import StandardDiffusionLoss
from .model import Model, Encoder, Decoder
from .model import Decoder, Encoder, Model
from .openaimodel import UNetModel
from .sampling import BaseDiffusionSampler
from .wrappers import OpenAIWrapper