Merge 6172e16572369879778c4a5a0f051de17c16283d into 2174ce5afea90ca489d222f539988dcef59f1027

This commit is contained in:
Quentin HENRY 2025-05-03 02:17:19 -04:00 committed by GitHub
commit f8812daa97
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: if sd_model is not None:
sd_unet.apply_unet("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) sd_hijack.model_hijack.undo_hijack(sd_model)
state_dict = get_checkpoint_state_dict(checkpoint_info, timer) state_dict = get_checkpoint_state_dict(checkpoint_info, timer)