Fix model_name_or_path
Browse files
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,
|