ibm-granite
/

granite-guardian-3.0-2b

Text Generation

Inference Endpoints

Model card Files Files and versions Community

ink-pad commited on 10 days ago

Commit

d40ae23

•

1 Parent(s): 692098a

Fix model_name_or_path

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -105,7 +105,7 @@ def get_probablities(logprobs):
     return probabilities
-model_path = "granite-guardian-3.0-2b"
 model = AutoModelForCausalLM.from_pretrained(
     model_path,

     return probabilities
+model_path = "ibm-granite/granite-guardian-3.0-2b"
 model = AutoModelForCausalLM.from_pretrained(
     model_path,