pankajmathur
/

model_101

Text Generation

text-generation-inference

Model card Files Files and versions Community

pankajmathur commited on Nov 9, 2023

Commit

7fde4e7

·

1 Parent(s): 6ec0b1d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -86,9 +86,9 @@ Below shows a code example on how to use this model
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-tokenizer = AutoTokenizer.from_pretrained("psmathur/model_007")
 model = AutoModelForCausalLM.from_pretrained(
-  "psmathur/model_007",
   torch_dtype=torch.float16,
   load_in_8bit=True,
   low_cpu_mem_usage=True,

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+tokenizer = AutoTokenizer.from_pretrained("pankajmathur/model_007")
 model = AutoModelForCausalLM.from_pretrained(
+  "pankajmathur/model_007",
   torch_dtype=torch.float16,
   load_in_8bit=True,
   low_cpu_mem_usage=True,