1

2026-03-18 11:18:49 +08:00
parent b3eda2f7c7
commit c8fb30aaee
3 changed files with 4 additions and 1 deletions
--- a/app/litserve_serve.py
+++ b/app/litserve_serve.py
@@ -37,6 +37,7 @@ class FluxKleinAPI(ls.LitAPI):

        self.pipe = Flux2KleinPipeline.from_pretrained(
            self.repo_id,
+            cache_dir="./checkpoint/",
            torch_dtype=self.dtype
        )
        self.pipe.to(device)