set_lora_device

app.py

@@ -29,30 +29,24 @@ def process_image(input_image, mode, weight1, weight2):
 sample_images = [
   'images/sample1.png',
   'images/sample2.png',
-  # 'images/sample3.png',
-  'images/sample4.png',
-  'images/sample5.png',
-  'images/sample6.png',
-  'images/sample7.png',
-  'images/sample8.png',
-  # 'images/sample9.png',
-  'images/sample10.png',
-  'images/sample11.png',
-  # 'images/sample12.png',
-  # 'images/sample13.png',
-  # 'images/sample14.png',
-  'images/sample15.png',
-  'images/sample16.png',
-  # 'images/sample17.png',
-  'images/sample18.png',
-  'images/sample19.png',
-  'images/sample20.png',
-  'images/sample21.png',
+#   'images/sample4.png',
+#   'images/sample5.png',
+#   'images/sample6.png',
+#   'images/sample7.png',
+#   'images/sample8.png',
+#   'images/sample10.png',
+#   'images/sample11.png',
+#   'images/sample15.png',
+#   'images/sample16.png',
+#   'images/sample18.png',
+#   'images/sample19.png',
+#   'images/sample20.png',
+#   'images/sample21.png',
 ]
 
 # Gradio インターフェースの定義
 with gr.Blocks() as demo:
-    gr.Markdown("# Image2Body Test")
+    gr.Markdown("# Image2Body Demo")
     
     with gr.Row():
         with gr.Column():

scripts/process_utils.py

@@ -102,7 +102,7 @@ def initialize_sotai_model():
     
     for lora_name, alpha in lora_names:
         lora_path = get_file_path(lora_name, subfolder=os.environ["lora_dir"])
-        load_lora(sotai_gen_pipe, lora_path, adapter_name=lora_name, alpha=alpha)
+        load_lora(sotai_gen_pipe, lora_path, adapter_name=lora_name.split(".")[0], alpha=alpha)
 
     # スケジューラーの設定
     sotai_gen_pipe.scheduler = UniPCMultistepScheduler.from_config(sotai_gen_pipe.scheduler.config)