Skip to content

Commit

Permalink
Diffusers transformers version update (#1029)
Browse files Browse the repository at this point in the history
* Skip saving pretrained model for diffuser.

* Skip saving pretrained model for diffuser, in nightly

* Skip checkpointing in hf-diffusers, hf-glue conv tests.

* Use the latest transformers

* Use transformers>=4.36.2
  • Loading branch information
yeounoh authored Dec 20, 2023
1 parent 5ecd8f3 commit 1d66e87
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/pytorch/nightly/hf-diffusers.libsonnet
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ local tpus = import 'templates/tpus.libsonnet';
cd examples/text_to_image
sed '/accelerate/d' requirements.txt > clean_requirements.txt
sed '/torchvision/d' requirements.txt > clean_requirements.txt
sed -i 's/transformers>=.*/transformers>=4.26.0/g' clean_requirements.txt
sed -i 's/transformers>=.*/transformers>=4.36.2/g' clean_requirements.txt
pip install -r clean_requirements.txt
# Skip saving the pretrained model, which contains invalid tensor storage
Expand Down

0 comments on commit 1d66e87

Please sign in to comment.