VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

initial commit for original U-Net

Browse Source
This commit is contained in:
Kohya S
2023-06-03 19:24:47 +09:00
parent 7c38c33ed6
commit 5db792b10b
3 changed files with 1238 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gen_img_diffusers.py
View File

@@ -78,7 +78,7 @@ from diffusers import (
HeunDiscreteScheduler,
KDPM2DiscreteScheduler,
KDPM2AncestralDiscreteScheduler,
UNet2DConditionModel,
# UNet2DConditionModel,
StableDiffusionPipeline,
)
from einops import rearrange
@@ -95,6 +95,7 @@ import library.train_util as train_util
from networks.lora import LoRANetwork
import tools.original_control_net as original_control_net
from tools.original_control_net import ControlNetInfo
from library.original_unet import UNet2DConditionModel
from XTI_hijack import unet_forward_XTI, downblock_forward_XTI, upblock_forward_XTI