Skip to content

Commit 3d30b7d

Browse files
Fix non-deterministic T5 outputs in HiDream pipeline tests (#13534)
avoid dropout of t5 model in hidream-image tests Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1 parent 62bfa5a commit 3d30b7d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/pipelines/hidream_image/test_pipeline_hidream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def get_dummy_components(self):
9696

9797
torch.manual_seed(0)
9898
config = AutoConfig.from_pretrained("hf-internal-testing/tiny-random-t5")
99-
text_encoder_3 = T5EncoderModel(config)
99+
text_encoder_3 = T5EncoderModel(config).eval()
100100

101101
torch.manual_seed(0)
102102
text_encoder_4 = LlamaForCausalLM.from_pretrained("hf-internal-testing/tiny-random-LlamaForCausalLM")

0 commit comments

Comments
 (0)