Upload folder using huggingface_hub
Browse files- handler.py +3 -2
handler.py
CHANGED
@@ -17,9 +17,10 @@ class EndpointHandler:
|
|
17 |
# self.model= load_model(path)
|
18 |
self.pipe = None
|
19 |
self.modelName = ""
|
|
|
20 |
baseReq = ImageRequest()
|
21 |
-
baseReq.model =
|
22 |
-
print(f"Loading model: {
|
23 |
self.LoadModel(baseReq)
|
24 |
|
25 |
def LoadModel(self, request):
|
|
|
17 |
# self.model= load_model(path)
|
18 |
self.pipe = None
|
19 |
self.modelName = ""
|
20 |
+
base = "sd-community/sdxl-flash"
|
21 |
baseReq = ImageRequest()
|
22 |
+
baseReq.model = base
|
23 |
+
print(f"Loading model: {base}")
|
24 |
self.LoadModel(baseReq)
|
25 |
|
26 |
def LoadModel(self, request):
|