Mohit imohitmayank

Lead Data Scientist

| AI/ML Researcher | Creator of "Jaal" | Author of "Lazy Data Science Guide" | TCS SuperCoder 2019

imohitmayank / training_llm_neural_codec_tts_model.py

Last active November 8, 2025 03:54

Training LLM-Based + Neural Codec TTS Models


	# %%
	# !pip install -q trl

	# %%
	import os
	os.environ["WANDB_PROJECT"] = "gemma3-snac-finetuning"
	os.environ["WANDB_LOG_MODEL"] = "checkpoint"
	os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"