waypoint-1-small

Paused

multimodalart HF Staff commited on Jan 22

Commit

404f154

verified ·

1 Parent(s): d995d67

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,13 +62,13 @@ pipe.to("cuda")
 pipe.blocks.sub_blocks['before_denoise'].sub_blocks['setup_kv_cache']._setup_kv_cache(pipe.transformer, pipe.device, torch.bfloat16)
 aoti_load_(
     pipe.transformer,
-    "diffusers-internal-dev/world-engine-aot",
     "transformer-fp8.pt2",
     "transformer-fp8-constants.pt"
 )
 aoti_load_(
     pipe.vae.decoder,
-    "diffusers-internal-dev/world-engine-aot",
     "decoder.pt2",
     "decoder-constants.pt"
 )

 pipe.blocks.sub_blocks['before_denoise'].sub_blocks['setup_kv_cache']._setup_kv_cache(pipe.transformer, pipe.device, torch.bfloat16)
 aoti_load_(
     pipe.transformer,
+    "diffusers/waypoint-1-small-aot",
     "transformer-fp8.pt2",
     "transformer-fp8-constants.pt"
 )
 aoti_load_(
     pipe.vae.decoder,
+    "diffusers/waypoint-1-small-aot",
     "decoder.pt2",
     "decoder-constants.pt"
 )