.

app.py

@@ -116,14 +116,17 @@ def generate_text_local(model_path, text):
     try:
         logger.info(f"Running local text generation with {model_path}")
         pipeline = pipelines[model_path]
-        pipeline.to("cuda")
+        
+        # デバイス情報をログに記録
+        device_info = next(pipeline.model.parameters()).device
+        logger.info(f"Model {model_path} is running on device: {device_info}")
+        
         outputs = pipeline(
             text,
             max_new_tokens=40,
             do_sample=False,
             num_return_sequences=1
         )
-        pipeline.to("cpu")
         return outputs[0]["generated_text"]
     except Exception as e:
         logger.error(f"Error in local text generation with {model_path}: {str(e)}")
@@ -149,9 +152,12 @@ def classify_text_local(model_path, text):
     try:
         logger.info(f"Running local classification with {model_path}")
         pipeline = pipelines[model_path]
-        pipeline.to("cuda")
+        
+        # デバイス情報をログに記録
+        device_info = next(pipeline.model.parameters()).device
+        logger.info(f"Model {model_path} is running on device: {device_info}")
+        
         result = pipeline(text)
-        pipeline.to("cpu")
         return str(result)
     except Exception as e:
         logger.error(f"Error in local classification with {model_path}: {str(e)}")