Commit
·
e305d87
1
Parent(s):
0c38620
Add the Riffusion checkpoint
Browse files- app.py +1 -6
- riffusion-model-v1.ckpt +3 -0
app.py
CHANGED
|
@@ -8,12 +8,7 @@ keras.mixed_precision.set_global_policy("mixed_float16")
|
|
| 8 |
resolution=512
|
| 9 |
sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights=False, jit_compile=True)
|
| 10 |
|
| 11 |
-
|
| 12 |
-
origin="https://huggingface.co/riffusion/riffusion-model-v1/resolve/main/riffusion-model-v1.ckpt",
|
| 13 |
-
file_hash="99a6eb51c18e16a6121180f3daa69344e571618b195533f67ae94be4eb135a57",
|
| 14 |
-
)
|
| 15 |
-
|
| 16 |
-
sd_dreambooth_model_1.load_weights_from_pytorch_ckpt(diffusion_model_pytorch_weights)
|
| 17 |
|
| 18 |
sd_dreambooth_model_1.diffusion_model.load_weights("/dreambooth_riffusion_model_currulao_v1")
|
| 19 |
|
|
|
|
| 8 |
resolution=512
|
| 9 |
sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights=False, jit_compile=True)
|
| 10 |
|
| 11 |
+
sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
sd_dreambooth_model_1.diffusion_model.load_weights("/dreambooth_riffusion_model_currulao_v1")
|
| 14 |
|
riffusion-model-v1.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99a6eb51c18e16a6121180f3daa69344e571618b195533f67ae94be4eb135a57
|
| 3 |
+
size 14580603928
|