Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -38,7 +38,7 @@ async def infer(model_index, prompt, nprompt="", height=0, width=0, steps=0, cfg
38
  if seed == -1: kwargs["seed"] = randomize_seed()
39
  else: kwargs["seed"] = seed
40
  task = asyncio.create_task(asyncio.to_thread(models2[model_index].fn,
41
- prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
42
  await asyncio.sleep(0)
43
  try:
44
  result = await asyncio.wait_for(task, timeout=timeout)
 
38
  if seed == -1: kwargs["seed"] = randomize_seed()
39
  else: kwargs["seed"] = seed
40
  task = asyncio.create_task(asyncio.to_thread(models2[model_index].fn,
41
+ prompt=prompt, negative_prompt=nprompt, **kwargs))
42
  await asyncio.sleep(0)
43
  try:
44
  result = await asyncio.wait_for(task, timeout=timeout)
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  transformers
2
  numpy<2
3
- torch==2.2.0
 
 
1
  transformers
2
  numpy<2
3
+ torch==2.2.0
4
+ huggingface_hub==0.26.2