gvij
/

SmolLM2-135M-Function-Calling

function-calling

edge-deployment

Model card Files Files and versions

gvij commited on 13 days ago

Commit

88edcdf

·

verified ·

1 Parent(s): 4f68028

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -67,7 +67,7 @@ This model is specifically designed for:
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "SmolLM2-135M-Function-Calling"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "gvij/SmolLM2-135M-Function-Calling"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 tokenizer = AutoTokenizer.from_pretrained(model_name)