davidberenstein1957 HF staff commited on
Commit
422059f
·
1 Parent(s): af420e8

update sft to chat

Browse files
src/synthetic_dataset_generator/app.py CHANGED
@@ -2,7 +2,7 @@ from synthetic_dataset_generator._tabbedinterface import TabbedInterface
2
 
3
  # from synthetic_dataset_generator.apps.eval import app as eval_app
4
  from synthetic_dataset_generator.apps.about import app as about_app
5
- from synthetic_dataset_generator.apps.sft import app as sft_app
6
  from synthetic_dataset_generator.apps.textcat import app as textcat_app
7
 
8
  theme = "argilla/argilla-theme"
@@ -21,8 +21,8 @@ button[role="tab"][aria-selected="true"]:hover {border-color: var(--button-prima
21
  image = """<br><img src="https://raw.githubusercontent.com/argilla-io/synthetic-data-generator/main/assets/logo.svg" alt="Synthetic Data Generator Logo" style="display: block; margin-left: auto; margin-right: auto; width: clamp(50%, 400px, 100%)"/>"""
22
 
23
  demo = TabbedInterface(
24
- [textcat_app, sft_app, about_app],
25
- ["Text Classification", "Supervised Fine-Tuning", "About"],
26
  css=css,
27
  title=image,
28
  theme=theme,
 
2
 
3
  # from synthetic_dataset_generator.apps.eval import app as eval_app
4
  from synthetic_dataset_generator.apps.about import app as about_app
5
+ from synthetic_dataset_generator.apps.chat import app as chat_app
6
  from synthetic_dataset_generator.apps.textcat import app as textcat_app
7
 
8
  theme = "argilla/argilla-theme"
 
21
  image = """<br><img src="https://raw.githubusercontent.com/argilla-io/synthetic-data-generator/main/assets/logo.svg" alt="Synthetic Data Generator Logo" style="display: block; margin-left: auto; margin-right: auto; width: clamp(50%, 400px, 100%)"/>"""
22
 
23
  demo = TabbedInterface(
24
+ [textcat_app, chat_app, about_app],
25
+ ["Text Classification", "Chat", "About"],
26
  css=css,
27
  title=image,
28
  theme=theme,
src/synthetic_dataset_generator/apps/{sft.py → chat.py} RENAMED
@@ -24,11 +24,7 @@ from synthetic_dataset_generator.constants import (
24
  MODEL,
25
  SFT_AVAILABLE,
26
  )
27
- from synthetic_dataset_generator.pipelines.embeddings import (
28
- get_embeddings,
29
- get_sentence_embedding_dimensions,
30
- )
31
- from synthetic_dataset_generator.pipelines.sft import (
32
  DEFAULT_DATASET_DESCRIPTIONS,
33
  generate_pipeline_code,
34
  get_magpie_generator,
@@ -36,6 +32,10 @@ from synthetic_dataset_generator.pipelines.sft import (
36
  get_prompt_rewriter,
37
  get_response_generator,
38
  )
 
 
 
 
39
  from synthetic_dataset_generator.utils import (
40
  get_argilla_client,
41
  get_org_dropdown,
 
24
  MODEL,
25
  SFT_AVAILABLE,
26
  )
27
+ from synthetic_dataset_generator.pipelines.chat import (
 
 
 
 
28
  DEFAULT_DATASET_DESCRIPTIONS,
29
  generate_pipeline_code,
30
  get_magpie_generator,
 
32
  get_prompt_rewriter,
33
  get_response_generator,
34
  )
35
+ from synthetic_dataset_generator.pipelines.embeddings import (
36
+ get_embeddings,
37
+ get_sentence_embedding_dimensions,
38
+ )
39
  from synthetic_dataset_generator.utils import (
40
  get_argilla_client,
41
  get_org_dropdown,
src/synthetic_dataset_generator/pipelines/{sft.py → chat.py} RENAMED
File without changes