Update app.py
Browse files
app.py
CHANGED
@@ -44,17 +44,7 @@ base_model = "black-forest-labs/FLUX.1-dev"
|
|
44 |
taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
|
45 |
good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
|
46 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, vae=taef1).to(device)
|
47 |
-
|
48 |
-
base_model,
|
49 |
-
vae=good_vae,
|
50 |
-
transformer=pipe.transformer,
|
51 |
-
text_encoder=pipe.text_encoder,
|
52 |
-
tokenizer=pipe.tokenizer,
|
53 |
-
text_encoder_2=pipe.text_encoder_2,
|
54 |
-
tokenizer_2=pipe.tokenizer_2,
|
55 |
-
torch_dtype=dtype
|
56 |
-
)
|
57 |
-
#multimodalart/flux-lora-lab - do not edit
|
58 |
MAX_SEED = 2**32 - 1
|
59 |
|
60 |
pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
|
|
|
44 |
taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
|
45 |
good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
|
46 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, vae=taef1).to(device)
|
47 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
MAX_SEED = 2**32 - 1
|
49 |
|
50 |
pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
|