search

mindspore-lab / mindone

one for all, Optimal generator with No Exception
Apache License 2.0
328 stars 62 forks source link

[Opensora-PKU v1.0]: change env exporting method & change default amp setting #536

Closed wtomin closed 2 weeks ago

wtomin commented 2 weeks ago

This PR refines the current Opensora-PKU v1.0 on the following ascepts:

  1. Previously we export environmental variables to switch between KBK and GE mode, which is ugly. Now we use an arguments to select which env file to be loaded.
  2. Change sample_t2v.py LatteT2V default data type from fp16 to bf16.