ozingmw
/

llama-3.2-3B-sentiment-kr-LoRA

Text Classification

Model card Files Files and versions Community

ozingmw commited on 17 days ago

Commit

4f54552

·

verified ·

1 Parent(s): 47d9a54

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ def load_model():
     global model, tokenizer, device
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    path = './llama-3.2-3B-sentiment_241105'
     tokenizer = AutoTokenizer.from_pretrained(path)
     model = AutoPeftModelForCausalLM.from_pretrained(

     global model, tokenizer, device
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    path = './llama-3.2-3B-sentiment-kr-LoRA'
     tokenizer = AutoTokenizer.from_pretrained(path)
     model = AutoPeftModelForCausalLM.from_pretrained(