Spaces:

p1atdev
/

TKG-Chroma-Key-AnimagineXL-4.0

Running on Zero

p1atdev commited on Aug 23

Commit

8e05eec

1 Parent(s): 939afee

fix: dtype

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def infer(
         std_dev=0.5,
         factor=8,
         channels=tkg_channels,
-    )
     images = pipe(
         latents=latents,

         std_dev=0.5,
         factor=8,
         channels=tkg_channels,
+    ).to(torch.bfloat16)
     images = pipe(
         latents=latents,