.

scripts/anime.py

@@ -14,8 +14,6 @@ from kornia.enhance import equalize_clahe
 from PIL import Image
 import numpy as np
 
-import spaces
-
 model = None
 device = None
 

scripts/generate_prompt.py

@@ -10,8 +10,6 @@ from tensorflow.keras.layers import TFSMLayer
 from huggingface_hub import hf_hub_download
 from pathlib import Path
 
-import spaces
-
 # 画像サイズの設定
 IMAGE_SIZE = 448
 

scripts/process_utils.py

@@ -10,12 +10,9 @@ from scripts.anime import init_model
 import torch
 from diffusers import StableDiffusionPipeline, StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler, AutoencoderKL
 import gc
-from peft import PeftModel
 from dotenv import load_dotenv
 from scripts.hf_utils import download_file
 
-import spaces
-
 # グローバル変数
 use_local = False
 model = None
@@ -46,9 +43,9 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
     print(f"\nDevice: {device}, Local model: {_use_local}\n")
 
     init_model(use_local)
-    # model = load_wd14_tagger_model()
+    model = load_wd14_tagger_model()
     sotai_gen_pipe = initialize_sotai_model()
-    # refine_gen_pipe = initialize_refine_model()
+    refine_gen_pipe = initialize_refine_model()
 
 def load_lora(pipeline, lora_path, adapter_name, alpha=0.75):
     pipeline.load_lora_weights(lora_path, adapter_name)