charliebaby2023 commited on
Commit
abbae1d
·
verified ·
1 Parent(s): 3c6753c

Update handle_models.py

Browse files
Files changed (1) hide show
  1. 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