Commit 41907b25 authored by Dalton's avatar Dalton Committed by GitHub

Cleanup sd_hijack_ddpm_v1.py

Forgot some things to revert
parent 4bc29633
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment