Update README.md

README.md

@@ -35,7 +35,7 @@ from datetime import datetime
 
 class GPUPredictions:
     def __init__(self,
-                 model_id="testmoto/gemma-2-llm2024-01",
+                 model_id="testmoto/gemma-2-lora-dpo-moe-1",
                  adapter_path=None,
                  max_tokens=1024,
                  temp=0.0,
@@ -215,7 +215,7 @@ class GPUPredictions:
 from GPUPredictions import GPUPredictions
 
 predictor = GPUPredictions(
-    model_id="testmoto/gemma-2-llm2024-01"
+    model_id="testmoto/gemma-2-lora-dpo-moe-1"
 )
 results = predictor.run_inference(
     input_file="elyza-tasks-100-TV_0.jsonl",