tori29umai0123

pretrained_model_name_or_path = "models/animagine-xl-3.1.safetensors"
train_data_dir = "train_data"
output_dir = "models/LoRA"
max_train_steps = 500
network_module = "networks.lora"
xformers = true
gradient_checkpointing = true
persistent_data_loader_workers = false
max_data_loader_n_workers = 0
enable_bucket = true

tori29umai0123

{
  "book": "Test",
  "pages": [
      {
          "page": 0,
          "width": 1654,
          "height": 1170,
          "texts": [
              {
                  "text": "こんにちは。いい天気ですね",

tori29umai0123

# 学習には　https://gist.github.com/tori29umai0123/fc51ea86aedc1b1e394b12829d1c95e5　のような形式のデータセットを利用する
# Manga109-sのアノテーションデータ（http://www.manga109.org/ja/annotations.html）を元に作成

import os
import glob
import json
import argparse
import torch
from torch.utils.data import Dataset, DataLoader
import torch.nn as nn

tori29umai0123

import configparser
import copy
import math
import os
import sys
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
from scipy.interpolate import PchipInterpolator 
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas

tori29umai0123

import os
import sys
import time
import socket
import gradio as gr
from llama_cpp import Llama
import datetime
from jinja2 import Template
import configparser
import threading

tori29umai0123

import csv
from pathlib import Path
import re
from sentence_transformers import SentenceTransformer
import faiss
from llama_cpp import Llama


import numpy as np

tori29umai0123

1girl,4114588
solo,3426446
highres,3008413
long_hair,2898315
commentary_request,2610959
breasts,2252741
looking_at_viewer,2017032
blush,1987494
smile,1847264
short_hair,1530601

tori29umai0123

import csv
from pathlib import Path
import re
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import create_retrieval_chain
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain_community.vectorstores import FAISS
from langchain_community.llms import LlamaCpp
from langchain.prompts import PromptTemplate
from langchain_core.documents import Document

tori29umai0123

import os
from concurrent.futures import ThreadPoolExecutor
from PIL import Image, ImageFilter

from PIL import Image, ImageFilter

def apply_blur_to_image(image_path, output_path, blur_radius):
    """指定された画像にブラー効果を適用し、指定されたパスに保存する"""
    try:
        image = Image.open(image_path)

tori29umai0123

#線画あり画像から、線画を取り除くスクリプト。線画を取り除きたい画像（image_dir）があれば動く。線画入力不要。

from PIL import Image, ImageFilter, ImageOps
from collections import defaultdict
from skimage import color as sk_color
from tqdm import tqdm
from skimage.color import deltaE_ciede2000, rgb2lab
import cv2
import numpy as np
import os