Kokoro-API-1

Sleeping

Yaron Koresh commited on Jan 16

Commit

d947d19

verified ·

1 Parent(s): a5c674b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ function custom(){
 # torch pipes
-image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
 video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     "THUDM/CogVideoX-5b-I2V",
     torch_dtype=dtype

 # torch pipes
+image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=dtype).to(device)
 video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     "THUDM/CogVideoX-5b-I2V",
     torch_dtype=dtype