TemplateAssertionError: no test named 'False'

[SaraTGRN]

I am trying to save SetFit in my directory (which is databricks), but both commands of :
trainer.model.save_pretrained(save directory)
OR trainer.save_pretrained(save directory)

gives me the error of TemplateAssertionError: no test named 'False',
can you help me how can I save my model.

[OliverXiaoxiong]

Sample problem, any solution?

[szhong-ssmhealth]

The following fixed the issue from my side when I use cpu for testing.
!pip install jinja2 --upgrade
import os
os.environ["TOKENIZERS_PARALLELISM"] = "false"

dump(trainer.model, 'setfit_model.pkl')
model=load('setfit_model.pkl')