Hugging Face's logo

Spaces:
rishiraj
/
dataset-chat-template
Running

App Files Community
rishiraj commited on
Commit
a229104
·
1 Parent(s): 8bd3596

restructure

Browse files
Files changed (5) hide show
  1. __init__.py → alignment/__init__.py +0 -0
  2. configs.py → alignment/configs.py +0 -0
  3. data.py → alignment/data.py +0 -0
  4. model_utils.py → alignment/model_utils.py +0 -0
  5. app.py +7 -1
__init__.py → alignment/__init__.py RENAMED
File without changes
configs.py → alignment/configs.py RENAMED
File without changes
data.py → alignment/data.py RENAMED
File without changes
model_utils.py → alignment/model_utils.py RENAMED
File without changes
app.py CHANGED
@@ -1,5 +1,11 @@
1
  import gradio as gr
2
- from . import DataArguments, ModelArguments, apply_chat_template, get_datasets, get_tokenizer
 
 
 
 
 
 
3
 
4
  def template(base_model, trained_adapter, token):
5
  data_args = DataArguments(chat_template=None, dataset_mixer={'HuggingFaceH4/no_robots': 1.0}, dataset_splits=['train_sft', 'test_sft'], max_train_samples=None, max_eval_samples=None, preprocessing_num_workers=12, truncation_side=None)
 
1
  import gradio as gr
2
+ from alignment import (
3
+ DataArguments,
4
+ ModelArguments,
5
+ apply_chat_template,
6
+ get_datasets,
7
+ get_tokenizer,
8
+ )
9
 
10
  def template(base_model, trained_adapter, token):
11
  data_args = DataArguments(chat_template=None, dataset_mixer={'HuggingFaceH4/no_robots': 1.0}, dataset_splits=['train_sft', 'test_sft'], max_train_samples=None, max_eval_samples=None, preprocessing_num_workers=12, truncation_side=None)