Update app.py

app.py

@@ -38,11 +38,11 @@ taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.b
 good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
 
 dtype = torch.bfloat16
-base_model = "AlekseyCalvin/FlexAlpha_Soonr"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, use_safetensors=True).to("cuda")
+pipe = DiffusionPipeline.from_pretrained("ostris/Flex.1-alpha", torch_dtype=dtype).to("cuda")
 #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 torch.cuda.empty_cache()
 
+pipe.transformer = FluxTransformer2DModel.from_pretrained("AlekseyCalvin/FlexAlpha_Soonr", use_safetensors=True)
 model_id = ("zer0int/LongCLIP-GmP-ViT-L-14")
 config = CLIPConfig.from_pretrained(model_id)
 config.text_config.max_position_embeddings = 248