fix from_single_file for checkpoints with t5 (huggingface#8631)

yiyixuxu · web-flow · commit 394b8fb99620 · 2024-06-19T08:23:35.000-10:00
fix single file
diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py
@@ -1808,3 +1808,4 @@ def create_diffusers_t5_model_from_checkpoint(
 
     else:
         model.load_state_dict(diffusers_format_checkpoint)
+    return model

Original file line number	Diff line number	Diff line change
`@@ -1808,3 +1808,4 @@ def create_diffusers_t5_model_from_checkpoint(`
`1808`	`1808`
`1809`	`1809`	`else:`
`1810`	`1810`	`model.load_state_dict(diffusers_format_checkpoint)`
	`1811`	`+ return model`