Spaces:
Sleeping
Sleeping
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
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|