DBCMLAB
/

Llama-3-instruction-constructionsafety-layertuning

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

juungwon commited on about 16 hours ago

Commit

d38c220

•

1 Parent(s): 3a5875d

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 = "DBCM/Llama-3-instruction-constructionsafety-layertuning"
 token = "your_access_token"
 tuned_model = AutoModelForCausalLM.from_pretrained(
     model_name,

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