allstax
/

CodeExplainer-7b-v0.1

Text Generation

text-generation-inference

8-bit precision

Model card Files Files and versions

allstax commited on Dec 16, 2023

Commit

9c20c6d

·

1 Parent(s): af0d69d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ Prompt Template for alpaca style
 # Loading the model
 ```
-from transformers import AutoModelForCausalLM, AutoTokenizer, GPTQConfig
 model_id = "allstax/CodeExplainer-7b-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 quant_model = AutoModelForCausalLM.from_pretrained(model_id, device_map='auto')

 # Loading the model
 ```
+from transformers import AutoModelForCausalLM, AutoTokenizer
 model_id = "allstax/CodeExplainer-7b-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 quant_model = AutoModelForCausalLM.from_pretrained(model_id, device_map='auto')