tori29umai0123

登場人物

勇者（男女選択可能）
数々の試練を乗り越え、魔王を討伐する勇者として認められた。
お人よし故に周囲に流されがちだが、いざという時は強い意志を見せる。
かつての幼馴染を取り戻すのが目的。

魔王（男女選択可能）
世界を破壊と混沌の闇に陥れようと目論む魔王。
勇者の幼馴染の体を奪い復活した。

tori29umai0123

import os
import sys
import numpy as np
import torch
from PIL import Image

sys.path.append(os.getcwd())  # 現在のディレクトリをシステムパスに追加

from tha3.poser.modes.load_poser import load_poser
from tha3.util import rgba_to_numpy_image, grid_change_to_numpy_image, rgb_to_numpy_image, extract_pytorch_image_from_PIL_image

tori29umai0123

import random
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from typing import Literal

# RATING、ASPECT_RATIO、LENGTH の選択肢を指定
RATING = Literal[
    "<|rating:general|>", 
    "<|rating:sensitive|>", 

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