This commit is contained in:
Quentin HENRY 2025-12-18 09:42:23 -05:00 committed by GitHub
commit b8e054db86
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -952,7 +952,7 @@ def reload_model_weights(sd_model=None, info=None, forced_reload=False):
if sd_model is not None:
sd_unet.apply_unet("None")
send_model_to_cpu(sd_model)
send_model_to_device(sd_model)
sd_hijack.model_hijack.undo_hijack(sd_model)
state_dict = get_checkpoint_state_dict(checkpoint_info, timer)