From f6916acbfa138f14e7c920d114fceae07c87939a Mon Sep 17 00:00:00 2001
From: nftblackmagic <108916776+nftblackmagic@users.noreply.github.com>
Date: Tue, 26 Nov 2024 12:21:01 -0800
Subject: [PATCH] Update app.py

---
 app.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 2a7c788..1368c03 100644
--- a/app.py
+++ b/app.py
@@ -17,12 +17,12 @@
 print('Loading diffusion model ...')
 transformer = FluxTransformer2DModel.from_pretrained(
     "xiaozaa/catvton-flux-alpha", 
-    torch_dtype=device
+    torch_dtype=dtype
 )
 pipe = FluxFillPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     transformer=transformer,
-    torch_dtype=device
+    torch_dtype=dtype
 ).to(device)
 print('Loading Finished!')