Cleanup sd_hijack_ddpm_v1.py

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

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