added models

rprasad2 · Jun 14, 2023 · 03546a5 · 03546a5
1 parent 0f2ff62
commit 03546a5
Show file tree

Hide file tree

Showing 4 changed files with 822 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -180,7 +180,7 @@ generated_imgs/
 
 # Custom model related artefacts
 variants.json
-models/
+# models/
 
 # models folder
 apps/stable_diffusion/web/models/

diff --git a/apps/stable_diffusion/src/models/__init__.py b/apps/stable_diffusion/src/models/__init__.py
@@ -0,0 +1,12 @@
+from apps.stable_diffusion.src.models.model_wrappers import (
+    SharkifyStableDiffusionModel,
+)
+from apps.stable_diffusion.src.models.opt_params import (
+    get_vae_encode,
+    get_vae,
+    get_unet,
+    get_clip,
+    get_tokenizer,
+    get_params,
+    get_variant_version,
+)