Kokoro-API-1

Sleeping

Yaron Koresh commited on Aug 23, 2024

Commit

8a757aa

verified ·

1 Parent(s): 688d8e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,7 +131,7 @@ pipe.scheduler = DDIMScheduler(
 pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
 pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
-pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device,dtype=dtype)
 # Parallelism

 pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
 pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
+pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device,dtype=dtype)
 # Parallelism