Update config.py

config.py

@@ -68,7 +68,7 @@ class Config:
     processor = AutoProcessor.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(phi_model_name, trust_remote_code=True)
     
-    projection = load_projection_model("models/MModalGPT-FINETUNE-step=3200-loss=1.01.ckpt", 768, 2560)
+    projection = load_projection_model("models/MModalGPT-FINETUNE-continued-step=10100-loss=1.16.ckpt", 768, 2560)
 
     clip_model = CLIPVisionModel.from_pretrained(model_name)
     audio_model = whisperx.load_model("small", device.type, compute_type="float16")
@@ -80,4 +80,4 @@ class Config:
                                                   return_dict=True,
                                                   trust_remote_code=True)
     
-    peft_model = peft.PeftModel.from_pretrained(text_model, 'models/3200')
+    peft_model = peft.PeftModel.from_pretrained(text_model, 'models/10100')