From 28f5a1b8be1607f831270e221ce71ea65b09957d Mon Sep 17 00:00:00 2001
From: Johannes Stelzer <stelzer@lunar-ring.ai>
Date: Mon, 9 Jan 2023 09:58:18 +0100
Subject: [PATCH] docstrings
---
stable_diffusion_holder.py | 15 ++++++++++++++-
1 file changed, 14 insertions(+), 1 deletion(-)
diff --git a/stable_diffusion_holder.py b/stable_diffusion_holder.py
index d76bc1e..c2696a9 100644
--- a/stable_diffusion_holder.py
+++ b/stable_diffusion_holder.py
@@ -116,7 +116,17 @@ class StableDiffusionHolder:
device: str = None,
precision: str='autocast',
):
-
+ r"""
+ Initializes the stable diffusion holder, which contains the models and sampler.
+ Args:
+ fp_ckpt: File pointer to the .ckpt model file
+ fp_config: File pointer to the .yaml config file
+ num_inference_steps: Number of diffusion iterations. Will be overwritten by latent blending.
+ height: Height of the resulting image.
+ width: Width of the resulting image.
+ device: Device to run the model on.
+ precision: Precision to run the model on.
+ """
self.seed = 42
self.guidance_scale = 5.0
@@ -148,6 +158,9 @@ class StableDiffusionHolder:
def init_model(self, fp_ckpt, fp_config):
+ r"""Loads the models and sampler.
+ """
+
assert os.path.isfile(fp_ckpt), f"Your model checkpoint file does not exist: {fp_ckpt}"
assert os.path.isfile(fp_config), f"Your config file does not exist: {fp_config}"
self.fp_ckpt = fp_ckpt