Hugging Face's logo

Spaces:
black-forest-labs
/
FLUX.1-dev
Running on Zero

App Files Community
229
multimodalart HF staff commited on
Commit
ffca250
•
1 Parent(s): 12c8016

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-large-patch14", torch
16
  text_encoder_2 = T5EncoderModel.from_pretrained(bfl_repo, subfolder="text_encoder_2", torch_dtype=dtype, revision="refs/pr/1")
17
  tokenizer_2 = T5TokenizerFast.from_pretrained(bfl_repo, subfolder="tokenizer_2", torch_dtype=dtype, revision="refs/pr/1")
18
  vae = AutoencoderKL.from_pretrained(bfl_repo, subfolder="vae", torch_dtype=dtype, revision="refs/pr/1")
19
- transformer = FluxTransformer2DModel.from_pretrained("diffusers-internal-dev/FLUX.1-dev", subfolder="transformer", torch_dtype=dtype)
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
 
 
16
  text_encoder_2 = T5EncoderModel.from_pretrained(bfl_repo, subfolder="text_encoder_2", torch_dtype=dtype, revision="refs/pr/1")
17
  tokenizer_2 = T5TokenizerFast.from_pretrained(bfl_repo, subfolder="tokenizer_2", torch_dtype=dtype, revision="refs/pr/1")
18
  vae = AutoencoderKL.from_pretrained(bfl_repo, subfolder="vae", torch_dtype=dtype, revision="refs/pr/1")
19
+ transformer = FluxTransformer2DModel.from_pretrained("diffusers-internal-dev/FLUX.1-dev", torch_dtype=dtype)
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22