Alina Lozovskaya commited on
Commit
cf69537
·
1 Parent(s): de5e2e6

Fix config

Browse files
Files changed (1) hide show
  1. yourbench_space/config.py +3 -5
yourbench_space/config.py CHANGED
@@ -18,7 +18,6 @@ def generate_base_config(
18
  "hf_organization": hf_org,
19
  "hf_dataset_name": hf_dataset_name,
20
  },
21
- "local_dataset_dir": "/app/results/",
22
  "model_list": [
23
  {
24
  "model_name": model_name,
@@ -38,13 +37,12 @@ def generate_base_config(
38
  },
39
  "pipeline": {
40
  "ingestion": {
41
- "source_documents_dir": "/app/example/raw",
42
- "output_dir": "/app/example/ingested",
43
  "run": True,
44
  },
45
  "upload_ingest_to_hub": {
46
- "source_documents_dir": "/app/example/ingested",
47
- "hub_dataset_name": f"{hf_dataset_name}_ingested_documents",
48
  "run": True,
49
  },
50
  "summarization": {"run": True},
 
18
  "hf_organization": hf_org,
19
  "hf_dataset_name": hf_dataset_name,
20
  },
 
21
  "model_list": [
22
  {
23
  "model_name": model_name,
 
37
  },
38
  "pipeline": {
39
  "ingestion": {
40
+ "source_documents_dir": "/app/uploaded_files",
41
+ "output_dir": "/app/ingested",
42
  "run": True,
43
  },
44
  "upload_ingest_to_hub": {
45
+ "source_documents_dir": "/app/ingested",
 
46
  "run": True,
47
  },
48
  "summarization": {"run": True},