.

app.py

@@ -9,9 +9,6 @@ from scripts.anime import init_model
 from datetime import datetime
 from pytz import timezone
 from scripts.survey import handle_form_submission, handle_visit_choice, handle_proceed, localize, script, generate_image, send_feedback
-import torch
-
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 初期化
 initialize(_use_local=False, use_gpu=True, use_dotenv=True)

scripts/process_utils.py

@@ -45,10 +45,10 @@ 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()
-    sotai_gen_pipe = initialize_sotai_model()
-    refine_gen_pipe = initialize_refine_model()
+    # init_model(use_local)
+    # model = load_wd14_tagger_model()
+    # sotai_gen_pipe = initialize_sotai_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)