Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ language:
|
|
14 |
## Model Information
|
15 |
|
16 |
|
17 |
-
# Fireball-R1-LLama-3.1-8B
|
18 |
|
19 |
 
|
20 |
|
@@ -57,8 +57,8 @@ pip install -U transformers
|
|
57 |
|
58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
59 |
|
60 |
-
tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Fireball-R1-Llama-3.1-8B")
|
61 |
-
model = AutoModelForCausalLM.from_pretrained("EpistemeAI/Fireball-R1-Llama-3.1-8B")
|
62 |
|
63 |
prompt = "Calculate the molar mass of sulfuric acid (H₂SO₄)."
|
64 |
inputs = tokenizer(prompt, return_tensors="pt")
|
@@ -71,7 +71,7 @@ import torch
|
|
71 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
72 |
|
73 |
# Load the tokenizer
|
74 |
-
tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Fireball-R1-Llama-3.1-8B")
|
75 |
|
76 |
# Load the model in 8-bit precision using bitsandbytes (requires a CUDA GPU)
|
77 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
14 |
## Model Information
|
15 |
|
16 |
|
17 |
+
# Fireball-R1.1-LLama-3.1-8B
|
18 |
|
19 |
 
|
20 |
|
|
|
57 |
|
58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
59 |
|
60 |
+
tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Fireball-R1.1-Llama-3.1-8B")
|
61 |
+
model = AutoModelForCausalLM.from_pretrained("EpistemeAI/Fireball-R1.1-Llama-3.1-8B")
|
62 |
|
63 |
prompt = "Calculate the molar mass of sulfuric acid (H₂SO₄)."
|
64 |
inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
71 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
72 |
|
73 |
# Load the tokenizer
|
74 |
+
tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Fireball-R1.1-Llama-3.1-8B")
|
75 |
|
76 |
# Load the model in 8-bit precision using bitsandbytes (requires a CUDA GPU)
|
77 |
model = AutoModelForCausalLM.from_pretrained(
|