Commit 4e6e2574 authored by Dalton's avatar Dalton Committed by GitHub

Cleanup ddpm_edit.py

Fully reverts this time
parent 41907b25
......@@ -9,12 +9,10 @@ https://github.com/CompVis/taming-transformers
# File modified by authors of InstructPix2Pix from original (https://github.com/CompVis/stable-diffusion).
# See more details in LICENSE.
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