VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +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, HeunDiscreteScheduler,
KDPM2DiscreteScheduler, KDPM2DiscreteScheduler,
KDPM2AncestralDiscreteScheduler, KDPM2AncestralDiscreteScheduler,
UNet2DConditionModel, # UNet2DConditionModel,
StableDiffusionPipeline, StableDiffusionPipeline,
) )
from einops import rearrange from einops import rearrange
@@ -95,6 +95,7 @@ import library.train_util as train_util
from networks.lora import LoRANetwork from networks.lora import LoRANetwork
import tools.original_control_net as original_control_net import tools.original_control_net as original_control_net
from tools.original_control_net import ControlNetInfo 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 from XTI_hijack import unet_forward_XTI, downblock_forward_XTI, upblock_forward_XTI

3
library/model_util.py
View File

@@ -5,8 +5,9 @@ import math
import os import os
import torch import torch
from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig, logging from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig, logging
from diffusers import AutoencoderKL, DDIMScheduler, StableDiffusionPipeline, UNet2DConditionModel from diffusers import AutoencoderKL, DDIMScheduler, StableDiffusionPipeline #, UNet2DConditionModel
from safetensors.torch import load_file, save_file from safetensors.torch import load_file, save_file
from library.original_unet import UNet2DConditionModel
# DiffUsers版StableDiffusionのモデルパラメータ # DiffUsers版StableDiffusionのモデルパラメータ
NUM_TRAIN_TIMESTEPS = 1000 NUM_TRAIN_TIMESTEPS = 1000

1234
library/original_unet.py Normal file
View File

File diff suppressed because it is too large Load Diff