Skip to content

Implementation of MagViT2 Tokenizer in Pytorch

License

Notifications You must be signed in to change notification settings

syyxsxx/magvit2-pytorch

 
 

Repository files navigation

install

pip install -r requirements.txt

python setup.py install

run

set device config

accelerate config

run train

accelerate launch main.py [dataset]

参数

  • --data-type (str): 训练数据类型,可选值[videos,images],默认videos
  • --bn (int): 训练batchsize,默认1

About

Implementation of MagViT2 Tokenizer in Pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%