Started diffusion

This commit is contained in:
Abhik Ahuja committed 2022-11-28 13:00:54 -08:00
1 parent 8f36e25510
commit 44fc1fc801
1 file changed
+15
+15
View File
@@ -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__()