Hugging Face's logo

Rustamshry
/
Llama3.2-SQL-1B

Text2Text Generation
PEFT
Safetensors
English
SQL
Text-to-SQL
SQL-generation
Model card Files Community
Rustamshry commited on
Commit
0318d07
·
verified ·
1 Parent(s): 7f807b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -61,7 +61,7 @@ login(token="")
61
  tokenizer = AutoTokenizer.from_pretrained("unsloth/Llama3.2-1B-Instruct",)
62
  base_model = AutoModelForCausalLM.from_pretrained(
63
  "unsloth/Llama3.2-1B-Instruct",
64
- device_map="auto", token=""
65
  )
66
 
67
  model = PeftModel.from_pretrained(base_model,"Rustamshry/Llama3.2-SQL-1B")
 
61
  tokenizer = AutoTokenizer.from_pretrained("unsloth/Llama3.2-1B-Instruct",)
62
  base_model = AutoModelForCausalLM.from_pretrained(
63
  "unsloth/Llama3.2-1B-Instruct",
64
+ device_map={"": 0}, token=""
65
  )
66
 
67
  model = PeftModel.from_pretrained(base_model,"Rustamshry/Llama3.2-SQL-1B")