初始化项目,由ModelHub XC社区提供模型
Model: mnoukhov/pythia410m-sft-tldr Source: Original Platform
This commit is contained in:
14
code/push_to_hub.py
Normal file
14
code/push_to_hub.py
Normal file
@@ -0,0 +1,14 @@
|
||||
# from transformers import AutoModelForCausalLM
|
||||
#
|
||||
#
|
||||
# name_or_path = "/home/toolkit/huggingface/tldr_sft_pythia7b_4V100_seq550"
|
||||
#
|
||||
# model = AutoModelForCausalLM.from_pretrained(name_or_path)
|
||||
#
|
||||
# model.push_to_hub("tldr_sft_pythia7b_4V100_seq550", use_temp_dir=False)
|
||||
|
||||
from datasets import load_from_disk
|
||||
|
||||
|
||||
ds = load_from_disk("/home/toolkit/huggingface/openai_summarize_comparisons_pythia1b")
|
||||
ds.push_to_hub("openai_summarize_comparisons_relabel_pythia1b")
|
||||
Reference in New Issue
Block a user