Update app.py

nftblackmagic · Nov 26, 2024 · f6916ac · f6916ac
1 parent 7b183da
commit f6916ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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!')