search

hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
https://hpcaitech.github.io/Open-Sora/
Apache License 2.0
21.77k stars 2.11k forks source link

Fix the CUDA out of memory when load pretrained model #421

Closed chehx closed 4 months ago

chehx commented 4 months ago

Please see the issue:https://github.com/hpcaitech/Open-Sora/issues/409

Owing to the config discrepancy, when you load pre-trained weight to train, it will forbid acceleration tech.

Fix via the code, or fix via huggingface config.