Merge pull request huggingface#1 from huggingface/add-glide

anton-l · web-flow · commit 0e6027d0ac77 · 2022-06-07T11:57:58.000+02:00
Add glide modeling files
diff --git a/__init__.py b/__init__.py
@@ -6,5 +6,6 @@
 
 from .modeling_utils import ModelMixin
 from .models.unet import UNetModel
+from .models.unet_glide import UNetGLIDEModel
 from .pipeline_utils import DiffusionPipeline
 from .schedulers.gaussian_ddpm import GaussianDDPMScheduler
diff --git a/models/__init__.py b/models/__init__.py
@@ -17,3 +17,4 @@
 # limitations under the License.
 
 from .unet import UNetModel
+from .unet_glide import UNetGLIDEModel
diff --git a/models/unet_glide.py b/models/unet_glide.py

Original file line number	Diff line number	Diff line change
`@@ -17,3 +17,4 @@`
`17`	`17`	`# limitations under the License.`
`18`	`18`
`19`	`19`	`from .unet import UNetModel`
	`20`	`+from .unet_glide import UNetGLIDEModel`