Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,8 @@ def port_inc():
|
|
30 |
os.environ["CUSTOM_PORT"]=str(int(env)+1)
|
31 |
|
32 |
def init_pool():
|
33 |
-
port_inc()
|
|
|
34 |
|
35 |
def pipe_t2i():
|
36 |
PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
|
|
|
30 |
os.environ["CUSTOM_PORT"]=str(int(env)+1)
|
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")
|