rishiraj commited on
Commit
a229104
β€’
1 Parent(s): 8bd3596

restructure

Browse files
__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)