Update lora.py

lora.py

@@ -16,17 +16,9 @@ from peft import PeftModel
 from snac import SNAC
 
 # ------------------------- CONFIG -------------------------
-
-LORA_REPO = "rahul7star/vaani-lora-lata"
-LORA_NAME="rahul7star/vaani-lora-lata"
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-BASE_MODEL = "rahul7star/nava1.0"   # Base model with tokenizer
-DATASET_HF = "rahul7star/vaani-snac-cleaned"
-SEQ_LEN = 512  # maximum text token length
-
-
-
-
+MODEL_NAME = "rahul7star/nava1.0"
+LORA_NAME = "rahul7star/vaani-lora-lata"
+SNAC_MODEL_NAME = "hubertsiuzdak/snac_24khz"
 TARGET_SR = 24000
 OUT_ROOT = Path("/tmp/data")
 OUT_ROOT.mkdir(exist_ok=True, parents=True)