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()
|