legolasyiu commited on
Commit
7a0269f
·
verified ·
1 Parent(s): d2bb905

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -14,7 +14,7 @@ language:
14
  ## Model Information
15
 
16
 
17
- # Fireball-R1-LLama-3.1-8B
18
 
19
  ![License](https://img.shields.io/badge/License-Apache%202.0-blue) ![Version](https://img.shields.io/badge/Version-1.0.0-green)
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
  ![License](https://img.shields.io/badge/License-Apache%202.0-blue) ![Version](https://img.shields.io/badge/Version-1.0.0-green)
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(