Spaces:

Culda
/

flux-inpaint-controlnet

Running

Culda commited on Sep 28, 2024

Commit

9f14500

1 Parent(s): 6c841a2

bfloat

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 #
 # login()
-dtype = torch.float16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"
@@ -47,7 +47,7 @@ def get_system_memory():
     }
-@spaces.GPU(duration=30)
 def inpaint(
     image,
     mask,

 #
 # login()
+dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"
     }
+@spaces.GPU(duration=60)
 def inpaint(
     image,
     mask,