OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-01 12:25:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup ddpm_edit.py

Browse Source 
Fully reverts this time
This commit is contained in:
Dalton 2024-03-20 23:36:35 -04:00 committed by GitHub
parent 41907b25f0
commit 4e6e2574ab
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
modules/models/diffusion/ddpm_edit.py
View File

@ -9,12 +9,10 @@ https://github.com/CompVis/taming-transformers
# File modified by authors of InstructPix2Pix from original (https://github.com/CompVis/stable-diffusion). # File modified by authors of InstructPix2Pix from original (https://github.com/CompVis/stable-diffusion).
# See more details in LICENSE. # See more details in LICENSE.
import sys
import torch import torch
import torch.nn as nn import torch.nn as nn
import numpy as np import numpy as np
import pytorch_lightning as pl import pytorch_lightning as pl
import pytorch_lightning.utilities.rank_zero
from torch.optim.lr_scheduler import LambdaLR from torch.optim.lr_scheduler import LambdaLR
from einops import rearrange, repeat from einops import rearrange, repeat
from contextlib import contextmanager from contextlib import contextmanager