Update handler.py
Browse files- 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:
|