Update app.py

app.py

@@ -38,13 +38,17 @@ with open('loras.json', 'r') as f:
 
 dtype = torch.bfloat16
 
-model = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", subfolder="transformer", torch_dtype=dtype).to("cuda")
-model.num_single_layers="0"
-model.chunk_size="0"
-model.pooled_projections="(_, _, 1)[0]"
-model.pooled_projections_dim="1"
+#model = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", subfolder="transformer", torch_dtype=dtype).to("cuda")
+#model.num_single_layers="0"
+#model.chunk_size="0"
+#model.pooled_projections="(_, 1)[0]"
+#model.pooled_projections_dim="0"
 pipe = FluxWithCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtype).to("cuda")
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
+#pipe.num_single_layers="0"
+pipe.transformer_chunk_size="0"
+#model.pooled_projections="(_, 1)[0]"
+pipe.transformer_pooled_projections_dim="0"
 
 pipe.to("cuda")
 #clipmodel = 'norm'