hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-09 13:49:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Formatting sd_hijack_ddpm_v1.py

Browse Source
This commit is contained in:
Dalton
2024-03-19 14:54:06 -04:00
committed by GitHub
parent 8f450321fe
commit 49779413aa
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/LDSR/sd_hijack_ddpm_v1.py
View File

@@ -9,13 +9,13 @@ 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
from functools import partial
from tqdm import tqdm
from torchvision.utils import make_grid
import pytorch_lightning.utilities.rank_zero
sys.modules['pytorch_lightning.utilities.distributed'] = sys.modules['pytorch_lightning.utilities.rank_zero']
from pytorch_lightning.utilities.rank_zero import rank_zero_only