Update app.py

app.py

@@ -42,17 +42,17 @@ JS = """function () {
 }"""
 
 
-# vae = AutoencoderKL.from_pretrained(
-#     repo,
-#     subfolder="vae",
-#     torch_dtype=torch.float16,
-# )
+vae = AutoencoderKL.from_pretrained(
+    repo,
+    subfolder="vae",
+    torch_dtype=torch.float16,
+ )
 
-# transformer = SD3Transformer2DModel.from_pretrained(
-#     repo, 
-#     subfolder="transformer",
-#     torch_dtype=torch.float16,
-# )
+transformer = SD3Transformer2DModel.from_pretrained(
+    repo, 
+    subfolder="transformer",
+    torch_dtype=torch.float16,
+)
 
 
 # text_encoder_3 = T5EncoderModel.from_pretrained(
@@ -71,9 +71,13 @@ JS = """function () {
 if torch.cuda.is_available():
     pipe = StableDiffusion3Pipeline.from_pretrained(
         repo,
+        vae=vae,
+        transformer=transformer,
         torch_dtype=torch.float16).to("cuda")
     pipe2 = StableDiffusion3Img2ImgPipeline.from_pretrained(
         repo,
+        vae=vae,
+        transformer=transformer,
         torch_dtype=torch.float16).to("cuda")