@spaces.GPU

app.py

@@ -8,11 +8,8 @@ from scripts.process_utils import initialize, process_image_as_base64
 from scripts.anime import init_model
 from scripts.generate_prompt import load_wd14_tagger_model
 
-# from dotenv import load_dotenv
-# load_dotenv()
-
 # 初期化
-initialize(_use_local=False, use_gpu=True)
+initialize(_use_local=False, use_gpu=True, use_dotenv=False)
 init_model(use_local=False)
 load_wd14_tagger_model()
 

scripts/process_utils.py

@@ -54,6 +54,7 @@ def load_lora(pipeline, lora_path, alpha=0.75):
 
 def initialize_sotai_model():
     global device, torch_dtype
+    print(f"Device: {device}, torch_dtype: {torch_dtype}")
 
     sotai_sd_model_path = get_file_path(os.environ["sotai_sd_model_name"], subfolder=os.environ["sd_models_dir"])
     controlnet_path1 =  get_file_path(os.environ["controlnet_name1"], subfolder=os.environ["controlnet_dir2"])