DBCMLAB
/

Llama-3-instruction-constructionsafety-layertuning

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

juungwon commited on May 28

Commit

fe53f77

•

1 Parent(s): 2fcead0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -34,7 +34,7 @@ After fine-tuning the entire layers, layers 0, 30, and 31 were replaced with par
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-model_name = "juungwon/Llama-3-instruction-constructionsafety-layertuning"
 tuned_model = AutoModelForCausalLM.from_pretrained(
     model_name,
     token=access_token,

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+model_name = "DBCM/Llama-3-instruction-constructionsafety-layertuning"
 tuned_model = AutoModelForCausalLM.from_pretrained(
     model_name,
     token=access_token,