mod allegro
This commit is contained in:
parent
c5e4fc68c9
commit
46e2c21cfd
19
allegro.py
19
allegro.py
|
|
@ -26,15 +26,19 @@ def fetch_documents():
|
|||
response = collection.query.fetch_objects()
|
||||
documents = []
|
||||
for item in response["data"]["Get"]["Document"]:
|
||||
file_name = item.get("fileName", "unknown_file")
|
||||
content = item.get("content", "")
|
||||
if content:
|
||||
documents.append(f"fileName: {file_name}, content: {content}")
|
||||
print(f"fileName: {file_name}")
|
||||
return documents
|
||||
print(item)
|
||||
#file_name = item.get("fileName", "unknown_file")
|
||||
#content = item.get("content", "")
|
||||
#if content:
|
||||
# documents.append(f"fileName: {file_name}, content: {content}")
|
||||
# print(f"fileName: {file_name}")
|
||||
#return documents
|
||||
#return documents
|
||||
|
||||
documents = fetch_documents()
|
||||
|
||||
fetch_documents()
|
||||
|
||||
"""
|
||||
# 3️⃣ Inicjalizacja modelu
|
||||
model_name = "allegro/multislav-5lang"
|
||||
device = "cuda" if torch.cuda.is_available() else "cpu"
|
||||
|
|
@ -99,3 +103,4 @@ model.save_pretrained("./models/allegro")
|
|||
tokenizer.save_pretrained("./models/allegro")
|
||||
|
||||
print("✅ Model został wytrenowany i zapisany!")
|
||||
"""
|
||||
Loading…
Reference in New Issue