ink-pad commited on
Commit
fc629b2
1 Parent(s): 4a08ed8

Fix model_name_or_path

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -104,7 +104,7 @@ def get_probablities(logprobs):
104
 
105
  return probabilities
106
 
107
- model_path = "granite-guardian-3.0-8b"
108
 
109
  model = AutoModelForCausalLM.from_pretrained(
110
  model_path,
 
104
 
105
  return probabilities
106
 
107
+ model_path = "ibm-granite/granite-guardian-3.0-8b"
108
 
109
  model = AutoModelForCausalLM.from_pretrained(
110
  model_path,