gokaygokay
/

SmolLM2-Prompt-Enhance

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

gokaygokay commited on 4 days ago

Commit

69ccb4c

•

1 Parent(s): bd60b4f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "gokaygokay/SmolLM2-135M-Instruct-Prompt-Enhance"
 tokenizer_id = "HuggingFaceTB/SmolLM2-135M-Instruct"
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_id )

 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_id = "gokaygokay/SmolLM2-Prompt-Enhance"
 tokenizer_id = "HuggingFaceTB/SmolLM2-135M-Instruct"
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_id )