Update README.md

README.md

@@ -1,5 +1,4 @@
 ---
-license: apache-2.0
 base_model: ibm-granite/granite-3.2-8b-instruct
 library_name: peft
 ---
@@ -46,7 +45,7 @@ device=torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 ANSWERABILITY_PROMPT = "<|start_of_role|>answerability<|end_of_role|>"
 BASE_NAME = "ibm-granite/granite-3.2-8b-instruct"
-LORA_NAME = "ibm-granite/granite-rag-answerability-prediction-3.2-8b-lora"
+LORA_NAME = "ibm-granite/granite-3.2-8b-lora-rag-answerability-prediction"
 
 tokenizer = AutoTokenizer.from_pretrained(BASE_NAME, padding_side='left',trust_remote_code=True)
 model_base = AutoModelForCausalLM.from_pretrained(BASE_NAME,device_map="auto")