Minor fix

app.py

@@ -67,18 +67,18 @@ def load_models():
             subfolder="unet"
         )
         
-        checkpoint_path = os.path.join(CHECKPOINT_DIR, "pytorch_model.bin")
-        if not os.path.exists(checkpoint_path):
+        unet_checkpoint_path = os.path.join(CHECKPOINT_DIR, "pytorch_model.bin")
+        if not os.path.exists(unet_checkpoint_path):
             print("⏳ Загрузка чекпоинта модели...")
-            hf_hub_download(
+            temp_path = hf_hub_download(
                 repo_id="Benrise/VITON-HD",
                 filename="VITONHD/model/pytorch_model.bin",
-                token=TOKEN,
-                local_dir=CHECKPOINT_DIR,
-                force_filename="pytorch_model.bin"
+                token=TOKEN
             )
-        
-        unet.load_state_dict(torch.load(checkpoint_path))
+            import shutil
+            shutil.copy(temp_path, unet_checkpoint_path)
+            print(f"✅ Файл успешно сохранён по пути: {unet_checkpoint_path}")
+        unet.load_state_dict(torch.load(unet_checkpoint_path, map_location=device))
         
         cloth_encoder = ClothEncoder.from_pretrained(
             "stabilityai/stable-diffusion-xl-base-1.0",