OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-03-09 15:34:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup sd_hijack_ddpm_v1.py

Browse Source 
Forgot some things to revert
This commit is contained in:
Dalton 2024-03-20 23:35:32 -04:00 committed by GitHub
parent 4bc2963320
commit 41907b25f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/LDSR/sd_hijack_ddpm_v1.py
View File

@ -4,12 +4,10 @@
# Some models such as LDSR require VQ to work correctly
# The classes are suffixed with "V1" and added back to the "ldm.models.diffusion.ddpm" module
import sys
import torch
import torch.nn as nn
import numpy as np
import pytorch_lightning as pl
import pytorch_lightning.utilities.rank_zero
from torch.optim.lr_scheduler import LambdaLR
from einops import rearrange, repeat
from contextlib import contextmanager