Colab crashes after using all avilable RAM

[Bruh-byte]

On the model loading step

[Jovonni]

This isn’t an issue with the repo, it’s your collab environment

Can you put the code for the cells crashing it?

[Bruh-byte]

This isn’t an issue with the repo, it’s your collab environment

Can you put the code for the cells crashing it?

model = "5b_lyrics" # or "1b_lyrics"
hps = Hyperparams()
hps.sr = 44100
hps.n_samples = 3 if model=='5b_lyrics' else 8
hps.name = 'samples'
chunk_size = 16 if model=="5b_lyrics" else 32
max_batch_size = 3 if model=="5b_lyrics" else 16
hps.levels = 3
hps.hop_fraction = [.5,.5,.125]

vqvae, *priors = MODELS[model]
vqvae = make_vqvae(setup_hparams(vqvae, dict(sample_length = 1048576)), device)
top_prior = make_prior(setup_hparams(priors[-1], dict()), vqvae, device)

[Bruh-byte]

Chose 1b version