Update README.md
Browse files
README.md
CHANGED
|
@@ -55,7 +55,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 55 |
print(f"Using device: {device}")
|
| 56 |
|
| 57 |
# Load model and tokenizer
|
| 58 |
-
model_path = "izzulgod/
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(model_path).to(device)
|
| 61 |
|
|
|
|
| 55 |
print(f"Using device: {device}")
|
| 56 |
|
| 57 |
# Load model and tokenizer
|
| 58 |
+
model_path = "izzulgod/gpt2-indo-instruct-tuned"
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(model_path).to(device)
|
| 61 |
|