fountai commited on
Commit
1bc7667
·
1 Parent(s): 35d3f47
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,9 +1,12 @@
 
 
 
 
 
1
  import spaces
2
  import gradio as gr
3
  from PIL import Image
4
  import os
5
- import torch.multiprocessing as mp
6
- mp.set_start_method('spawn', force=True)
7
 
8
  from src.flux.xflux_pipeline import XFluxPipeline
9
  import random
 
1
+ import torch.multiprocessing as mp
2
+ import multiprocessing as mp2
3
+ mp.set_start_method('spawn', force=True)
4
+ mp2.set_start_method('spawn', force=True)
5
+
6
  import spaces
7
  import gradio as gr
8
  from PIL import Image
9
  import os
 
 
10
 
11
  from src.flux.xflux_pipeline import XFluxPipeline
12
  import random