POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod allegro

This commit is contained in:
l.gabrysiak 2025-02-28 21:19:50 +01:00
parent 2d82373bc8
commit 04747ff17b
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
allegro.py
View File

@ -36,12 +36,8 @@ def fetch_documents():
return documents return documents
#return documents #return documents
documents = fetch_documents()
fetch_documents()
client.close()
"""
# 3️⃣ Inicjalizacja modelu # 3️⃣ Inicjalizacja modelu
model_name = "allegro/multislav-5lang" model_name = "allegro/multislav-5lang"
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
@ -105,5 +101,4 @@ trainer.train()
model.save_pretrained("./models/allegro") model.save_pretrained("./models/allegro")
tokenizer.save_pretrained("./models/allegro") tokenizer.save_pretrained("./models/allegro")
print("✅ Model został wytrenowany i zapisany!") print("✅ Model został wytrenowany i zapisany!")
"""