Update app.py

app.py

@@ -42,7 +42,6 @@ pipe = FluxWithCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtyp
 
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")
 
-
 pipe.to("cuda")
 clipmodel = 'norm'
 if clipmodel == "long":
@@ -55,13 +54,16 @@ if clipmodel == "norm":
     maxtokens = 512
 clip_model = CLIPModel.from_pretrained(model_id, torch_dtype=torch.bfloat16, config=config, ignore_mismatched_sizes=False).to("cuda")
 clip_processor = CLIPProcessor.from_pretrained(model_id, padding="max_length", max_length=maxtokens, ignore_mismatched_sizes=False, return_tensors="pt", truncation=True)
-pipe.transformer = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", num_single_layers=0, chunk_size=0)
+pipe.transformer = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", subfolder="transformer", num_single_layers=0, chunk_size=0)
 pipe.tokenizer = clip_processor.tokenizer
 pipe.text_encoder = clip_model.text_model
 pipe.tokenizer_max_length = maxtokens
 pipe.text_encoder.dtype = torch.bfloat16
 torch.cuda.empty_cache()
 
+num_single_layers=0 
+chunk_size=0
+
 MAX_SEED = 2**32-1
 
 class calculateDuration: