Commit
·
d987e13
1
Parent(s):
3435252
update examples
Browse files
examples/enforce_mapgie_template copy.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
-
os.environ["MAGPIE_PRE_QUERY_TEMPLATE"] = "
|
7 |
-
os.environ["MODEL"] = "
|
8 |
|
9 |
launch()
|
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
+
os.environ["MAGPIE_PRE_QUERY_TEMPLATE"] = "my_custom_template"
|
7 |
+
os.environ["MODEL"] = "google/gemma-2-9b-it"
|
8 |
|
9 |
launch()
|
examples/ollama_local.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
-
assert os.getenv("HF_TOKEN")
|
7 |
os.environ["BASE_URL"] = "http://127.0.0.1:11434/v1/"
|
8 |
os.environ["MODEL"] = "llama3.1"
|
9 |
|
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
+
assert os.getenv("HF_TOKEN") # push the data to huggingface
|
7 |
os.environ["BASE_URL"] = "http://127.0.0.1:11434/v1/"
|
8 |
os.environ["MODEL"] = "llama3.1"
|
9 |
|
examples/openai_local.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
-
assert os.getenv("HF_TOKEN")
|
7 |
os.environ["BASE_URL"] = "https://api.openai.com/v1/"
|
8 |
os.environ["API_KEY"] = os.getenv("OPENAI_API_KEY")
|
9 |
os.environ["MODEL"] = "gpt-4o"
|
|
|
3 |
|
4 |
from synthetic_dataset_generator import launch
|
5 |
|
6 |
+
assert os.getenv("HF_TOKEN") # push the data to huggingface
|
7 |
os.environ["BASE_URL"] = "https://api.openai.com/v1/"
|
8 |
os.environ["API_KEY"] = os.getenv("OPENAI_API_KEY")
|
9 |
os.environ["MODEL"] = "gpt-4o"
|