Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def port_inc():
|
|
31 |
|
32 |
def init_pool():
|
33 |
#port_inc()
|
34 |
-
|
35 |
|
36 |
def pipe_t2i():
|
37 |
PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
|
@@ -242,7 +242,7 @@ if __name__ == "__main__":
|
|
242 |
pp2=pipe_i2i()
|
243 |
result=[]
|
244 |
|
245 |
-
mp.set_start_method("
|
246 |
|
247 |
port_inc()
|
248 |
ui()
|
|
|
31 |
|
32 |
def init_pool():
|
33 |
#port_inc()
|
34 |
+
print("init pool")
|
35 |
|
36 |
def pipe_t2i():
|
37 |
PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
|
|
|
242 |
pp2=pipe_i2i()
|
243 |
result=[]
|
244 |
|
245 |
+
mp.set_start_method("fork", force=True)
|
246 |
|
247 |
port_inc()
|
248 |
ui()
|