SalmanFaroz
/

Llama-2-7b-samsum

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

SalmanFaroz commited on Apr 24

Commit

81443ff

•

1 Parent(s): 9a53e00

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ You can use this model for text summarization tasks by utilizing the Hugging Fac
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
-model_id =  "SalmanFaroz/Llama-2-7b-QLoRa-samsum"
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
+model_id =  "SalmanFaroz/Llama-2-7b-samsum"
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,