Add the Riffusion checkpoint

app.py

@@ -8,12 +8,7 @@ keras.mixed_precision.set_global_policy("mixed_float16")
 resolution=512
 sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights=False, jit_compile=True)
 
-diffusion_model_pytorch_weights = keras.utils.get_file(
-    origin="https://huggingface.co/riffusion/riffusion-model-v1/resolve/main/riffusion-model-v1.ckpt",
-    file_hash="99a6eb51c18e16a6121180f3daa69344e571618b195533f67ae94be4eb135a57",
-)
-
-sd_dreambooth_model_1.load_weights_from_pytorch_ckpt(diffusion_model_pytorch_weights)
+sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
 
 sd_dreambooth_model_1.diffusion_model.load_weights("/dreambooth_riffusion_model_currulao_v1")
 

riffusion-model-v1.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99a6eb51c18e16a6121180f3daa69344e571618b195533f67ae94be4eb135a57
+size 14580603928