Sandiago21
/

llama-13b-hf-prompt-answering

Text Generation

decapoda-research-13b-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

13e1339

•

1 Parent(s): a40d15f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

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

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