Update app.py

app.py

@@ -303,9 +303,10 @@ class CharacterMaker:
         self.chat_history = []
         self.settings = None
         self.model_loaded = threading.Event()
-        self.current_model = None  # 現在ロードされているモデルのタイプを記録
+        self.current_model = None
         self.model_lock = threading.Lock()
         self.use_chat_format = False
+        self.last_loaded_settings = None
 
     def load_model(self, model_type):
         with self.model_lock:
@@ -313,36 +314,46 @@ class CharacterMaker:
             model_path = os.path.join(MODEL_DIR, self.settings[f'DEFAULT_{model_type.upper()}_MODEL'])
             n_gpu_layers = self.settings[f'{model_type.lower()}_n_gpu_layers']
 
-            # 現在のモデルが既にロードされているか、同じ設定であるかチェック
-            if (self.llama and self.current_model != model_type and
-                self.llama.model_path == model_path and
-                self.llama.n_gpu_layers == n_gpu_layers):
-                print(f"同じ設定のため {self.current_model} から {model_type} へのモデルタイプ変更を行いますが、再ロードはしません。")
-                self.current_model = model_type
-                return  # 設定が同じ場合は再ロードをスキップ
-
-            if (self.llama and
-                self.llama.model_path == model_path and
-                self.llama.n_gpu_layers == n_gpu_layers):
-                print(f"{model_type} モデルは既に同じ設定でロードされています。")
-                return  # 設定が同じ場合は再ロードをスキップ
+            # 設定が変更されたかどうかをチェック
+            current_settings = self.get_current_settings(model_type)
+            if self.last_loaded_settings != current_settings:
+                print(f"設定が変更されたため、{model_type}モデルを再ロードします。")
+                self.reload_model(model_type, model_path, n_gpu_layers)
+            elif self.llama and self.current_model == model_type:
+                print(f"{model_type}モデルは既にロードされており、設定も変更されていません。")
+                return
+            else:
+                print(f"{model_type}モデルをロードします。")
+                self.reload_model(model_type, model_path, n_gpu_layers)
 
-            # モデルの初期化
-            if self.llama:
-                del self.llama
-                self.llama = None
+    def reload_model(self, model_type, model_path, n_gpu_layers):
+        if self.llama:
+            del self.llama
+            self.llama = None
 
-            self.model_loaded.clear()
+        self.model_loaded.clear()
 
-            try:
-                # 新しいモデルをロード
-                self.llama = LlamaAdapter(model_path, params, n_gpu_layers)
-                self.current_model = model_type
-                self.model_loaded.set()
-                print(f"{model_type} モデルをロードしました。モデルパス: {model_path}、GPUレイヤー数: {n_gpu_layers}")
-            except Exception as e:
-                print(f"{model_type} モデルのロード中にエラーが発生しました: {e}")
-                self.model_loaded.set()
+        try:
+            self.llama = LlamaAdapter(model_path, params, n_gpu_layers)
+            self.current_model = model_type
+            self.model_loaded.set()
+            self.last_loaded_settings = self.get_current_settings(model_type)
+            print(f"{model_type}モデルをロードしました。モデルパス: {model_path}、GPUレイヤー数: {n_gpu_layers}")
+        except Exception as e:
+            print(f"{model_type}モデルのロード中にエラーが発生しました: {e}")
+            self.model_loaded.set()
+
+    def get_current_settings(self, model_type):
+        # 現在の設定を取得
+        return {
+            'model_path': self.settings[f'DEFAULT_{model_type.upper()}_MODEL'],
+            'n_gpu_layers': self.settings[f'{model_type.lower()}_n_gpu_layers'],
+            'temperature': self.settings[f'{model_type.lower()}_temperature'],
+            'top_p': self.settings[f'{model_type.lower()}_top_p'],
+            'top_k': self.settings[f'{model_type.lower()}_top_k'],
+            'rep_pen': self.settings[f'{model_type.lower()}_rep_pen'],
+            'n_ctx': self.settings[f'{model_type.lower()}_n_ctx']
+        }
 
 
     def generate_response(self, input_str):
@@ -621,14 +632,14 @@ def apply_settings():
             int(new_config['GenerateParameters'].get('n_ctx', '10000'))
         )
     
-    # モデルを再ロード
+    # モデルを再ロードするためにcurrent_modelをリセット
     character_maker.current_model = None
+    character_maker.last_loaded_settings = None
     
     # temp_settings をクリア
     temp_settings.clear()
     
     return "設定をiniファイルに保存し、アプリケーションに反映しました。次回の操作時に新しいモデルがロードされ���す。"
-
 # Gradioインターフェース
 def build_gradio_interface():
     global temp_settings