Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 23 |
print(f"Using device: {device}")
|
| 24 |
|
| 25 |
model_name = "jeff-vincent/distilgpt2-python-codegen"
|
| 26 |
-
output_model_path = "./codegen_distilgpt2-5"
|
| 27 |
|
| 28 |
# Load the tokenizer and model for causal language modeling
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 23 |
print(f"Using device: {device}")
|
| 24 |
|
| 25 |
model_name = "jeff-vincent/distilgpt2-python-codegen"
|
|
|
|
| 26 |
|
| 27 |
# Load the tokenizer and model for causal language modeling
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|