seedboxai
/

Llama-3-KafkaLM-8B-v0.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

doubledsbv commited on Jun 4, 2024

Commit

1312cb6

·

verified ·

1 Parent(s): 7d9ca23

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -50,7 +50,7 @@ import transformers
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_id = "seedboxai/Llama-3-Kafka-8B-v0.1"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_id = "seedboxai/Llama-3-KafkaLM-8B-v0.1"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)