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 22:41:35 +01:00
parent cd5fab2206
commit e3a94fa5ae
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
allegro.py
View File

@ -1,18 +1,11 @@
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
from safetensors.torch import load_file # Załaduj plik safetensors
# Załaduj tokenizer
tokenizer = AutoTokenizer.from_pretrained("./models/ably")
# Załaduj model z formatu safetensors
model = AutoModelForSeq2SeqLM.from_pretrained("ścieżka_do_modelu", from_tf=False, from_safetensors=True)
tokenizer = AutoTokenizer.from_pretrained("ścieżka_do_modelu")
# Załaduj model z pliku safetensors
model = AutoModelForSeq2SeqLM.from_pretrained("./models/ably", local_files_only=True)
# Zapisz model i tokenizer w formacie kompatybilnym z PyTorch (plik .bin)
model.save_pretrained("./models/ably")
tokenizer.save_pretrained("./models/ably")
# Załaduj model.wagi z pliku safetensors
model_weights = load_file("./models/ably/model.safetensors")
model.load_state_dict(model_weights)
# Zapisz model w formacie binarnym .bin dla PyTorch
model.save_pretrained("./models/ably_bin")
tokenizer.save_pretrained("./models/ably_bin")
print("✅ Model został zapisany w formacie binarnym!")
print("✅ Model zapisany w formacie binarnym!")