From 44fc1fc801d5392034266f922f0cda582a4ad3f2 Mon Sep 17 00:00:00 2001 From: Abhik Ahuja Date: Mon, 28 Nov 2022 13:00:54 -0800 Subject: [PATCH] Started diffusion --- nerfstudio/generative/stable_diffusion.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 nerfstudio/generative/stable_diffusion.py diff --git a/nerfstudio/generative/stable_diffusion.py b/nerfstudio/generative/stable_diffusion.py new file mode 100644 index 000000000..346a3b5c4 --- /dev/null +++ b/nerfstudio/generative/stable_diffusion.py @@ -0,0 +1,15 @@ +import torch +from diffusers import AutoencoderKL, PNDMScheduler, UNet2DConditionModel +from torch import nn +from transformers import CLIPTextModel, CLIPTokenizer, logging + + +class StableDiffusion(nn.Module): + """Stable Diffusion implementation + """ + + def __init__( + self, + device + ) -> None: + super().__init__()