Keltezaa commited on
Commit
ddecc54
Β·
verified Β·
1 Parent(s): 1a77a75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -11
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
- pipe_i2i = AutoPipelineForImage2Image.from_pretrained(
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)