Files
sbv2-v2/webui/inference.py
tsukumi 9e5222619f Refactor: No preloading of BERT models to avoid unnecessary GPU VRAM consumption during training in the Web UI
Since the BERT features of the dataset are pre-extracted by bert_gen.py, there is no need to load the BERT model at training time.
2024-03-13 12:09:11 +09:00

21 KiB