Skip to content

Commit e967294

Browse files
LaurentLaurent2916
authored andcommitted
update juggernaut aftermath weights to be float16 instead of float32
1 parent a5bc44f commit e967294

File tree

4 files changed

+9
-8
lines changed

4 files changed

+9
-8
lines changed

src/refiners/conversion/cli.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ def main() -> None:
7676
unet_sd15.runwayml_inpainting.convert()
7777
unet_sd15.juggernaut_reborn.convert()
7878
unet_sd15.juggernaut_aftermath.convert()
79+
unet_sd15.juggernaut_aftermath_inpainting.convert()
7980
unet_sd15.realistic_stock_photo_v3.convert()
8081
unet_sd15.realistic_vision_v5.convert()
8182

src/refiners/conversion/models/autoencoder_sd15.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -337,10 +337,10 @@
337337
converted=Hub(
338338
repo_id="refiners/sd15.juggernaut.aftermath.autoencoder",
339339
filename="model.safetensors",
340-
expected_sha256="fa7ef415e1854907aa46d81ed403bd1f6b0fd2f06c885545ab689f9f78fa7989",
340+
expected_sha256="83d147dd0d25231e49ba35820f7dd7f0902ba6e93237d00b6caee0599f010618",
341341
),
342342
recipe=civitai_recipe,
343-
dtype=torch.float32,
343+
dtype=torch.float16,
344344
)
345345
# autoencoder wise, juggernaut_aftermath = juggernaut_aftermath_inpainting
346346
realistic_stock_photo_v3 = Conversion(

src/refiners/conversion/models/clip_text_sd15.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,10 +254,10 @@
254254
converted=Hub(
255255
repo_id="refiners/sd15.juggernaut.aftermath.text_encoder",
256256
filename="model.safetensors",
257-
expected_sha256="2448c3148ae1c9658c18b136e5d6eebcc3c512dd3e1df71d3cb9bb1c83c19db1",
257+
expected_sha256="388ba94024540d450c32c6bdaa05a76c8f7ef42f6aef478b2e95a49220b1fd66",
258258
),
259259
recipe=civitai_recipe,
260-
dtype=torch.float32,
260+
dtype=torch.float16,
261261
)
262262
realistic_stock_photo_v3 = Conversion(
263263
original=Hub(

src/refiners/conversion/models/unet_sd15.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -850,10 +850,10 @@
850850
converted=Hub(
851851
repo_id="refiners/sd15.juggernaut.aftermath.unet",
852852
filename="model.safetensors",
853-
expected_sha256="5882c517eac0670df60755cb9eb762081a1d6a37431d3f7f9a3a6d8dfe764d86",
853+
expected_sha256="60312c91cb1807f9e1e1801b0686d9492fcea2b54835c7934b7aa433db0878e2",
854854
),
855855
recipe=civitai_recipe,
856-
dtype=torch.float32,
856+
dtype=torch.float16,
857857
)
858858
juggernaut_aftermath_inpainting = Conversion(
859859
original=Hub(
@@ -865,10 +865,10 @@
865865
converted=Hub(
866866
repo_id="refiners/sd15.juggernaut.aftermath.unet_inpainting",
867867
filename="model.safetensors",
868-
expected_sha256="c00a85060de351d617cc3a3be6865cf3493a0557f1f32303cb8a385c6368b9a8",
868+
expected_sha256="6fc035c3cb865ec1efe72005ff7a70d8605616c64ca68466e161c8a570bdab87",
869869
),
870870
recipe=civitai_recipe,
871-
dtype=torch.float32,
871+
dtype=torch.float16,
872872
)
873873
realistic_stock_photo_v3 = Conversion(
874874
original=Hub(

0 commit comments

Comments
 (0)