Skip to content

Commit

Permalink
[hf-diffusers] Use the latest transformers for diffusers (#1027)
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
  • Loading branch information
yeounoh authored Dec 11, 2023
1 parent a635d67 commit 3a3911f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/pytorch/nightly/hf-diffusers.libsonnet
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +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
pip install -r clean_requirements.txt
# Skip saving the pretrained model, which contains invalid tensor storage
Expand Down

0 comments on commit 3a3911f

Please sign in to comment.