Spaces:

gokaygokay
/

FLUX.1-dev-with-Captioner

Running on Zero

gokaygokay commited on Aug 3

Commit

ac745b0

•

1 Parent(s): 68036b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ dtype = torch.bfloat16
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 # FLUX.1-dev model
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype).to(device)
 # Initialize Florence model
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()

 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 # FLUX.1-dev model
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, token = huggingface_token).to(device)
 # Initialize Florence model
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()