Hugging Face's logo

Spaces:
zerogpu-aoti
/
Flux-Compiled-Graph
Running on Zero

App Files Community
2
sayakpaul HF Staff commited on
Commit
13352a6
·
1 Parent(s): 6580845

up

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -12,9 +12,8 @@ dtype = torch.bfloat16
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Load the model pipeline
15
- pipe = DiffusionPipeline.from_pretrained(
16
- "black-forest-labs/Flux.1-Dev", torch_dtype=dtype
17
- ).to(device)
18
 
19
  @spaces.GPU(duration=1500)
20
  def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken, progress=gr.Progress(track_tqdm=True)):
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Load the model pipeline
15
+ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/Flux.1-Dev", torch_dtype=dtype).to(device)
16
+
 
17
 
18
  @spaces.GPU(duration=1500)
19
  def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken, progress=gr.Progress(track_tqdm=True)):