nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 05:13:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #154 from hpcaitech/hotfix/t5_assert

Browse source 
fix assertion
This commit is contained in:
Shen Chenhui 2024-06-19 21:22:51 +08:00 committed by GitHub
commit 49536bd923
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
opensora/models/text_encoder/t5.py
View file

@ -99,7 +99,7 @@ class T5Embedder:
self.use_text_preprocessing = use_text_preprocessing
self.hf_token = hf_token
assert from_pretrained in self.available_models
# assert from_pretrained in self.available_models
self.tokenizer = AutoTokenizer.from_pretrained(
from_pretrained,
cache_dir=cache_dir,