Mirza Učanbarlić commited on
Commit ·
510acad
1
Parent(s): b841118
add weight name
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -10,7 +10,8 @@ class EndpointHandler():
|
|
| 10 |
def __init__(self, path=""):
|
| 11 |
repo_id = "runwayml/stable-diffusion-v1-5"
|
| 12 |
self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id)
|
| 13 |
-
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
|
|
|
| 10 |
def __init__(self, path=""):
|
| 11 |
repo_id = "runwayml/stable-diffusion-v1-5"
|
| 12 |
self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id)
|
| 13 |
+
weight_name = "pixel-portrait-v1.safetensors"
|
| 14 |
+
self.pipeline.load_lora_weights("simulationcartridge/ppl", weight_name=weight_name)
|
| 15 |
|
| 16 |
|
| 17 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|