Skip to content

GitLab

Unverified Commit 394b8fb9 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix from_single_file for checkpoints with t5 (#8631) 

fix single file
parent a1d55e14
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/loaders/single_file_utils.py
View file @ 394b8fb9
...@@ -1808,3 +1808,4 @@ def create_diffusers_t5_model_from_checkpoint( ...@@ -1808,3 +1808,4 @@ def create_diffusers_t5_model_from_checkpoint(
else: else:
model.load_state_dict(diffusers_format_checkpoint) model.load_state_dict(diffusers_format_checkpoint)
return model
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment