Skip to content

S
Stable Diffusion Webui
Commit 41907b25 authored by Dalton's avatar Dalton Committed by GitHub
Browse files
Options

Cleanup sd_hijack_ddpm_v1.py 

Forgot some things to revert
parent 4bc29633
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
extensions-builtin/LDSR/sd_hijack_ddpm_v1.py
View file @ 41907b25
...@@ -4,12 +4,10 @@ ...@@ -4,12 +4,10 @@
# Some models such as LDSR require VQ to work correctly # 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 # The classes are suffixed with "V1" and added back to the "ldm.models.diffusion.ddpm" module
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
......
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