Update inference_manager.py

inference_manager.py

@@ -18,6 +18,7 @@ import base64
 import json
 import jwt
 import glob
+import traceback
 
 #from onediffx import compile_pipe, save_pipe, load_pipe
 
@@ -153,7 +154,7 @@ class InferenceManager:
             
             #vae = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir, "vae"), torch_dtype=torch.bfloat16)
             vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.bfloat16)
-            
+            print(ckpt_dir)
             pipe = DiffusionPipeline.from_pretrained(
                 ckpt_dir,
                 vae=vae,
@@ -287,7 +288,7 @@ class ModelManager:
             return
 
         for file_path in model_files:
-            model_name = self.get_model_name_from_url(file_path)
+            model_name = self.get_model_name_from_url(file_path).split(".")[0]
             print(f"Initializing model: {model_name} from {file_path}")
             try:
                 # Initialize InferenceManager for each model
@@ -324,6 +325,7 @@ class ModelManager:
             print(f"Building pipeline with LoRAs for model {model_id}...")
             return model.build_pipeline_with_lora(lora_list, sampler, new_pipeline)
         except Exception as e:
+            traceback.print_exc()
             print(f"Failed to build pipeline for model {model_id}: {e}")
             return None