Update finetunning.py

finetunning.py

@@ -1,3 +1,4 @@
+import os
 import torch
 from datasets import Dataset # , load_dataset
 from transformers import (
@@ -8,6 +9,14 @@ from transformers import (
     DataCollatorForLanguageModeling
 )
 
+# Створюємо директорії з правами на запис для всіх користувачів
+os.makedirs('./cache', exist_ok=True)
+os.chmod('./cache', 0o777)
+os.makedirs('./results', exist_ok=True)
+os.chmod('./results', 0o777)
+os.makedirs('./fine_tuned_model', exist_ok=True)
+os.chmod('./fine_tuned_model', 0o777)
+
 # Завантаження моделі та токенізатора
 model_name = "facebook/opt-350m"
 tokenizer = AutoTokenizer.from_pretrained(model_name)