Kokoro-API-3

Running

Yaron Koresh commited on Jan 22

Commit

4e42c4b

verified ·

1 Parent(s): 6a56eb5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -513,7 +513,7 @@ function custom(){
 # torch pipes
 image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
-#image_pipe.enable_model_cpu_offload()
 torch.cuda.empty_cache()

 # torch pipes
 image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
+image_pipe.enable_model_cpu_offload()
 torch.cuda.empty_cache()