Commit
·
675b21b
1
Parent(s):
7389893
handler update
Browse files- __pycache__/handler.cpython-39.pyc +0 -0
- handler.py +2 -2
__pycache__/handler.cpython-39.pyc
CHANGED
|
Binary files a/__pycache__/handler.cpython-39.pyc and b/__pycache__/handler.cpython-39.pyc differ
|
|
|
handler.py
CHANGED
|
@@ -4,8 +4,8 @@ from diffusers import DiffusionPipeline
|
|
| 4 |
class EndpointHandler():
|
| 5 |
def __init__(self, path=""):
|
| 6 |
# Preload all the elements you are going to need at inference.
|
| 7 |
-
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
|
| 8 |
-
|
| 9 |
|
| 10 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
| 11 |
"""
|
|
|
|
| 4 |
class EndpointHandler():
|
| 5 |
def __init__(self, path=""):
|
| 6 |
# Preload all the elements you are going to need at inference.
|
| 7 |
+
self.pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
|
| 8 |
+
self.pipeline.load_lora_weights("jonathanrstern/logo-generator")
|
| 9 |
|
| 10 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
| 11 |
"""
|