evolution_pipeline.py

from diffusers.utils.testing_utils import enable_full_determinism
from EvoMusic.application import EvoMusic

import torch
import numpy as np
import random
import argparse
import warnings

if __name__ == "__main__":
    warnings.filterwarnings("ignore")
    enable_full_determinism()
    # create argparser
    parser = argparse.ArgumentParser(description="Evolve your own personalized music")
    parser.add_argument(
        "--config_path",
        type=str,
        default="config.yaml",
        help="Path to the configuration file",
    )
    # seed
    parser.add_argument(
        "--seed", type=int, default=42, help="Seed for reproducibility"
    )
    config_path = parser.parse_args().config_path
    # Set random seed for reproducibility
    seed = parser.parse_args().seed
    torch.manual_seed(seed)
    np.random.seed(seed)
    random.seed(seed)
    torch.cuda.seed_all()

    app = EvoMusic(config_path)

    app.generation_loop(0)