Texttra commited on
Commit
eaf5244
·
verified ·
1 Parent(s): 17119de

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -0
handler.py CHANGED
@@ -15,9 +15,14 @@ class EndpointHandler:
15
  torch_dtype=torch.float16,
16
  )
17
 
 
 
 
18
  # Load your local LoRA file
19
  try:
20
  lora_path = "./Bh0r1.safetensors" # relative path within the container
 
 
21
  self.pipe.load_lora_weights(lora_path)
22
  print(f"✅ LoRA weights loaded from {lora_path}.")
23
  except Exception as e:
 
15
  torch_dtype=torch.float16,
16
  )
17
 
18
+ # Diagnostic print to inspect available methods
19
+ print("🔍 Available methods on pipeline:", dir(self.pipe))
20
+
21
  # Load your local LoRA file
22
  try:
23
  lora_path = "./Bh0r1.safetensors" # relative path within the container
24
+
25
+ # Attempt to load LoRA weights (will fail if unsupported)
26
  self.pipe.load_lora_weights(lora_path)
27
  print(f"✅ LoRA weights loaded from {lora_path}.")
28
  except Exception as e: