philschmid
/

llama-2-7b-instruction-generator

Text Generation

text-generation-inference

Model card Files Files and versions

YoungPhlo commited on Sep 8, 2023

Commit

480f414

·

1 Parent(s): c8263f5

fix AutoModelForCausalLM import

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -49,10 +49,10 @@ July 25, 2023
 ```python
 import torch
-from transformers import AutoTokenizer, AutoModelModelForCausalLM
 # load base LLM model and tokenizer
-model = AutoModelModelForCausalLM.from_pretrained(
     "philschmid/llama-2-7b-instruction-generator",
     low_cpu_mem_usage=True,
     torch_dtype=torch.float16,

 ```python
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
 # load base LLM model and tokenizer
+model = AutoModelForCausalLM.from_pretrained(
     "philschmid/llama-2-7b-instruction-generator",
     low_cpu_mem_usage=True,
     torch_dtype=torch.float16,