Spaces:
Running
Running
Update handle_models.py
Browse files- handle_models.py +1 -0
handle_models.py
CHANGED
@@ -27,6 +27,7 @@ def load_fn(models,HF_TOKEN):
|
|
27 |
models_load.update({model: m})
|
28 |
|
29 |
async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1, timeout=inference_timeout):
|
|
|
30 |
kwargs = {}
|
31 |
if height > 0: kwargs["height"] = height
|
32 |
if width > 0: kwargs["width"] = width
|
|
|
27 |
models_load.update({model: m})
|
28 |
|
29 |
async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1, timeout=inference_timeout):
|
30 |
+
print(f"{prompt}")
|
31 |
kwargs = {}
|
32 |
if height > 0: kwargs["height"] = height
|
33 |
if width > 0: kwargs["width"] = width
|