raicrits
/

Llama3_ChangeOfTopic

Text2Text Generation

transformers, peft

Model card Files Files and versions Community

stefanoscotta commited on Jun 4

Commit

e3deb06

•

1 Parent(s): 7e6f4ed

Update README.md

Files changed (1) hide show

README.md +1 -3

README.md CHANGED Viewed

@@ -60,10 +60,8 @@ Use the code below to get started with the model.
  **Usage:**
 Use the code below to get started with the model.
  ``` python
-import os
 import torch
-import sys
-from transformers import LlamaForCausalLM, AutoTokenizer
 from peft import PeftModel
 model_id = "meta-llama/Meta-Llama-3-8B"

  **Usage:**
 Use the code below to get started with the model.
  ``` python
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 from peft import PeftModel
 model_id = "meta-llama/Meta-Llama-3-8B"