gabrielpondc
/

NL2SQL-StarCoder-15B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

gabrielpondc commited on Apr 16

Commit

1e3849a

•

1 Parent(s): 1991883

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -47,8 +47,8 @@ CREATE TABLE "table_name" (
 ```python
 import torch
-from modelscope import snapshot_download, AutoModelForCausalLM, AutoTokenizer,GenerationConfig
-model_dir = snapshot_download("iic/NL2SQL-StarCoder-15B")
 tokenizer = AutoTokenizer.from_pretrained(model_dir, device_map="auto",
                               trust_remote_code=True, torch_dtype=torch.float16)
 tokenizer.padding_side = "left"

 ```python
 import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+model_dir = "gabrielpondc/NL2SQL-StarCoder-15B"
 tokenizer = AutoTokenizer.from_pretrained(model_dir, device_map="auto",
                               trust_remote_code=True, torch_dtype=torch.float16)
 tokenizer.padding_side = "left"