Spaces:
Sleeping
Sleeping
Commit
β’
8250642
1
Parent(s):
aa69d47
Upload 6 files
Browse files
fypmc20277423/.gitattributes
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
fypmc20277423/README.md
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
title: Fypmc20277423
|
3 |
+
emoji: π
|
4 |
+
colorFrom: indigo
|
5 |
+
colorTo: gray
|
6 |
+
sdk: streamlit
|
7 |
+
sdk_version: 1.32.0
|
8 |
+
app_file: app.py
|
9 |
+
pinned: false
|
10 |
+
---
|
11 |
+
|
12 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
fypmc20277423/app.py
ADDED
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import GPT2Tokenizer, GPT2LMHeadModel, Trainer, TrainingArguments
|
2 |
+
from datasets import load_dataset
|
3 |
+
|
4 |
+
# Load the text dataset from the specified file.
|
5 |
+
dataset = load_dataset("text", data_files="training.txt")
|
6 |
+
|
7 |
+
# Initialize the GPT-2 tokenizer.
|
8 |
+
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
9 |
+
# Set the tokenizer's pad token to the EOS token.
|
10 |
+
tokenizer.pad_token = tokenizer.eos_token
|
11 |
+
|
12 |
+
# Define a function to tokenize the dataset and prepare labels.
|
13 |
+
def tokenize_function(examples):
|
14 |
+
# Tokenize the text to input_ids, attention_mask
|
15 |
+
tokenized_inputs = tokenizer(examples["text"], padding="max_length", truncation=True, max_length=512)
|
16 |
+
# Prepare labels: labels are the same as input_ids for language modeling
|
17 |
+
tokenized_inputs["labels"] = tokenized_inputs["input_ids"].copy()
|
18 |
+
return tokenized_inputs
|
19 |
+
|
20 |
+
# Tokenize the entire dataset.
|
21 |
+
tokenized_datasets = dataset.map(tokenize_function, batched=True)
|
22 |
+
# Remove the 'text' column as it's no longer needed after tokenization.
|
23 |
+
tokenized_datasets = tokenized_datasets.remove_columns(["text"])
|
24 |
+
# Set the format of the dataset to PyTorch tensors.
|
25 |
+
tokenized_datasets.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
|
26 |
+
|
27 |
+
# Load the GPT-2 model.
|
28 |
+
model = GPT2LMHeadModel.from_pretrained("gpt2")
|
29 |
+
|
30 |
+
# Define training arguments.
|
31 |
+
training_args = TrainingArguments(
|
32 |
+
output_dir="./output",
|
33 |
+
overwrite_output_dir=True,
|
34 |
+
num_train_epochs=3,
|
35 |
+
per_device_train_batch_size=4,
|
36 |
+
save_steps=10_000,
|
37 |
+
save_total_limit=2,
|
38 |
+
)
|
39 |
+
|
40 |
+
# Initialize the Trainer with the training dataset including labels.
|
41 |
+
trainer = Trainer(
|
42 |
+
model=model,
|
43 |
+
args=training_args,
|
44 |
+
train_dataset=tokenized_datasets["train"],
|
45 |
+
)
|
46 |
+
|
47 |
+
# Start the training process.
|
48 |
+
trainer.train()
|
49 |
+
|
50 |
+
# Save the fine-tuned model and tokenizer.
|
51 |
+
model.save_pretrained("fine_tuned_gpt2_model")
|
52 |
+
tokenizer.save_pretrained("fine_tuned_gpt2_model")
|
fypmc20277423/output/runs/Mar31_21-01-46_LAPTOP-23I4320M/events.out.tfevents.1711915307.LAPTOP-23I4320M.22972.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7f422cc424360ad0b6d022f492ce2d8d45fa8fb317ad09588d920344d1bfdaad
|
3 |
+
size 4805
|
fypmc20277423/requirements.txt
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
torch
|
2 |
+
transformers[torch]
|
3 |
+
streamlit
|
4 |
+
accelerate>=0.21.0
|
fypmc20277423/training.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|