setting default checkpoint to hosted model

2022-12-12 22:16:45 -08:00 · 2022-12-12 22:16:45 -08:00 · 37d975cea8
parent afa49af2b0
commit 37d975cea8
1 changed files with 1 additions and 1 deletions
--- a/riffusion/server.py
+++ b/riffusion/server.py
@ -44,7 +44,7 @@ SEED_IMAGES_DIR = Path(Path(__file__).resolve().parent.parent, "seed_images")

 def run_app(
    *,
-    checkpoint: str,
+    checkpoint: str = "riffusion/riffusion-model-v1",
    host: str = "127.0.0.1",
    port: int = 3000,
    debug: bool = False,