Skip to content

Commit

Permalink
update examples
Browse files Browse the repository at this point in the history
  • Loading branch information
davidberenstein1957 committed Dec 17, 2024
1 parent 3435252 commit d987e13
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions examples/enforce_mapgie_template copy.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from synthetic_dataset_generator import launch

os.environ["MAGPIE_PRE_QUERY_TEMPLATE"] = "llama3"
os.environ["MODEL"] = "my_custom_model_trained_on_llama3"
os.environ["MAGPIE_PRE_QUERY_TEMPLATE"] = "my_custom_template"
os.environ["MODEL"] = "google/gemma-2-9b-it"

launch()
2 changes: 1 addition & 1 deletion examples/ollama_local.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from synthetic_dataset_generator import launch

assert os.getenv("HF_TOKEN")
assert os.getenv("HF_TOKEN") # push the data to huggingface
os.environ["BASE_URL"] = "http://127.0.0.1:11434/v1/"
os.environ["MODEL"] = "llama3.1"

Expand Down
2 changes: 1 addition & 1 deletion examples/openai_local.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from synthetic_dataset_generator import launch

assert os.getenv("HF_TOKEN")
assert os.getenv("HF_TOKEN") # push the data to huggingface
os.environ["BASE_URL"] = "https://api.openai.com/v1/"
os.environ["API_KEY"] = os.getenv("OPENAI_API_KEY")
os.environ["MODEL"] = "gpt-4o"
Expand Down

0 comments on commit d987e13

Please sign in to comment.