mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 04:13:57 +08:00
Started diffusion
This commit is contained in:
1 parent
8f36e25510
commit
44fc1fc801
1 file changed
+15
@@ -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__()
|
||||
Reference in new issue
Block a user