Sandiago21
/

llama-7b-hf-prompt-answering

Text Generation

decapoda-research-7b-hf

prompt answering

text-generation-inference

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

Sandiago21 commited on Jun 10, 2023

Commit

e3f04e2

•

1 Parent(s): 8089fb8

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -100,6 +100,7 @@ Use the code below to get started with the model.
 ```python
 import torch
 from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
 MODEL_NAME = "Sandiago21/llama-7b-hf-prompt-answering"
@@ -157,6 +158,7 @@ print(response)
 ```python
 import torch
 from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
 MODEL_NAME = "Sandiago21/llama-7b-hf-prompt-answering"

 ```python
 import torch
+from peft import PeftConfig
 from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
 MODEL_NAME = "Sandiago21/llama-7b-hf-prompt-answering"
 ```python
 import torch
+from peft import PeftConfig
 from transformers import GenerationConfig, LlamaTokenizer, LlamaForCausalLM
 MODEL_NAME = "Sandiago21/llama-7b-hf-prompt-answering"