Spaces:
Running
Running
Upload 2 files
#9
by
John6666
- opened
- app.py +1 -1
- 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
|
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
|