Update README.md
Browse files
README.md
CHANGED
|
@@ -88,7 +88,6 @@ The model was fine-tuned on a combination of the following datasets:
|
|
| 88 |
import torch
|
| 89 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 90 |
|
| 91 |
-
# Set device (RTX 8000 with 48 GB memory)
|
| 92 |
device = "cuda"
|
| 93 |
|
| 94 |
# Load the merged fine-tuned model and tokenizer
|
|
|
|
| 88 |
import torch
|
| 89 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 90 |
|
|
|
|
| 91 |
device = "cuda"
|
| 92 |
|
| 93 |
# Load the merged fine-tuned model and tokenizer
|