mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-09-20 03:15:26 +00:00
12 lines
349 B
Python
12 lines
349 B
Python
|
from refiners.training_utils.latent_diffusion import FinetuneLatentDiffusionConfig, LatentDiffusionTrainer
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import sys
|
||
|
|
|
||
|
|
config_path = sys.argv[1]
|
||
|
|
config = FinetuneLatentDiffusionConfig.load_from_toml(
|
||
|
|
toml_path=config_path,
|
||
|
|
)
|
||
|
|
trainer = LatentDiffusionTrainer(config=config)
|
||
|
|
trainer.train()
|