Yntec John6666 commited on
Commit
e99671f
·
verified ·
1 Parent(s): 437578a

Upload 2 files (#9)

Browse files

- Upload 2 files (39c7258a161b3a40ff77a4f469fdee8c984984c8)


Co-authored-by: John Smith <[email protected]>

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