update

app.py

@@ -426,7 +426,7 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
                         
                         with gr.Row():
                             seed = gr.Number(
-                                label="Seed", value=649151, precision=0,
+                                label="Seed", value=-1, precision=0,
                                 info="Random seed for reproducibility. (-1 for random seed)"
                             )
                             use_reprompt = gr.Checkbox(
@@ -434,8 +434,8 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
                                 info="Enhance prompt automatically"
                             )
                             use_refiner = gr.Checkbox(
-                                label="Use Refiner", value=False,
-                                info="Apply refiner after generation (Refiner is not supported yet; coming soon.)",
+                                label="Use Refiner", value=True,
+                                info="Apply refiner after generation",
                                 interactive=True
                             )
                         

hyimage/models/vae/__init__.py

@@ -8,7 +8,11 @@ def load_vae(device, vae_path: str = None, vae_precision: str = None):
     config = HunyuanVAE2D.load_config(vae_path)
     vae = HunyuanVAE2D.from_config(config)
 
-    ckpt = torch.load(Path(vae_path) / "pytorch_model.ckpt", map_location='cpu')
+    ckpt_path = Path(vae_path) / "pytorch_model.ckpt"
+    if not ckpt_path.exists():
+        ckpt_path = Path(vae_path) / "pytorch_model.pt"
+
+    ckpt = torch.load(ckpt_path, map_location='cpu')
     if "state_dict" in ckpt:
         ckpt = ckpt["state_dict"]
     vae_ckpt = {}