Yaron Koresh commited on
Commit
967c4fc
·
verified ·
1 Parent(s): 8c20dad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ def port_inc():
31
 
32
  def init_pool():
33
  #port_inc()
34
- return
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("spawn", force=True)
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()