Update app.py

app.py

@@ -13,6 +13,11 @@ weight_dtype = torch.float32
 weight_dtype_te = torch.bfloat16
 MAX_SEED = np.iinfo(np.int32).max
 
+transformer = Transformer2DModel.from_pretrained(
+    "alfredplpl/commonart-tmp-3",
+    torch_dtype=weight_dtype
+)
+
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=weight_dtype)
 scheduler=DPMSolverMultistepScheduler()
 pipe = PixArtSigmaPipeline(