Commit
·
27fc45c
1
Parent(s):
2f0b7b2
second parameter
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -51,7 +51,8 @@
|
|
| 51 |
# dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
|
| 52 |
|
| 53 |
class EndpointHandler():
|
| 54 |
-
def __init__(self):
|
|
|
|
| 55 |
pass
|
| 56 |
# face_adapter = f"./checkpoints/ip-adapter.bin"
|
| 57 |
# controlnet_path = f"./checkpoints/ControlNetModel"
|
|
|
|
| 51 |
# dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
|
| 52 |
|
| 53 |
class EndpointHandler():
|
| 54 |
+
def __init__(self, model_dir):
|
| 55 |
+
print("Model dir: ", model_dir)
|
| 56 |
pass
|
| 57 |
# face_adapter = f"./checkpoints/ip-adapter.bin"
|
| 58 |
# controlnet_path = f"./checkpoints/ControlNetModel"
|