Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -135,7 +135,7 @@ pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_d
|
|
135 |
# Parallelism
|
136 |
|
137 |
def parallel(func,*args):
|
138 |
-
with Pool(
|
139 |
res = pool.map(func, *args)
|
140 |
return list(res)
|
141 |
|
|
|
135 |
# Parallelism
|
136 |
|
137 |
def parallel(func,*args):
|
138 |
+
with Pool(len(args)) as pool:
|
139 |
res = pool.map(func, *args)
|
140 |
return list(res)
|
141 |
|