multimodalart HF Staff commited on
Commit
32069c8
·
verified ·
1 Parent(s): c96ea4c

[Admin maintenance] Support new ZeroGPU hardware

Browse files

Thank you so much for having shared this Space with the community on this demo. We have upgraded the ZeroGPU infra-structure to run on modern blackwell architecture.
For that, we need to upgrade your demo to support that. This PR fixes your demo to work with the new architecture. As this is something we broke on our end, we may merge this PR autonomously. If this breaks unexpectedly or brings unintended consequences, feel free to revert, modify or otherwise. Any issues you can email apolinario@huggingface.co

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from diffusers import DiffusionPipeline
6
  import torch
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
- model_repo_id = "models/black-forest-labs/FLUX.1-schnell" #Replace to the model you would like to use
10
 
11
  if torch.cuda.is_available():
12
  torch_dtype = torch.float16
 
6
  import torch
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
+ model_repo_id = "black-forest-labs/FLUX.1-schnell" #Replace to the model you would like to use
10
 
11
  if torch.cuda.is_available():
12
  torch_dtype = torch.float16