Yaron Koresh commited on
Commit
163cfaf
·
verified ·
1 Parent(s): 695d112

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -25,13 +25,12 @@ def port_inc():
25
  os.environ["GRADIO_SERVER_PORT"]=str(int(os.getenv("GRADIO_SERVER_PORT"))+1)
26
 
27
  def init_pool(q):
28
- port_inc()
29
  global b
30
  b = q
31
 
32
  _dynamo.config.suppress_errors = True
33
 
34
- mp.set_start_method("spawn", force=True)
35
  b=mp.Queue()
36
 
37
  #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1", token=os.getenv("hf_token")).to(device)
 
25
  os.environ["GRADIO_SERVER_PORT"]=str(int(os.getenv("GRADIO_SERVER_PORT"))+1)
26
 
27
  def init_pool(q):
 
28
  global b
29
  b = q
30
 
31
  _dynamo.config.suppress_errors = True
32
 
33
+ mp.set_start_method("fork", force=True)
34
  b=mp.Queue()
35
 
36
  #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1", token=os.getenv("hf_token")).to(device)