Skip to content

Commit

Permalink
fix: force input dataset name and model name
Browse files Browse the repository at this point in the history
  • Loading branch information
ledong0110 committed Aug 13, 2024
1 parent 569d82f commit 16cfa63
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/vieval/script_arguments.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,14 +4,14 @@

@dataclass
class ScriptArguments:
model_name: Optional[str] = field(
model_name: str = field(
default="meta-llama/Llama-2-7b-chat-hf",
metadata={
"help": "The model that you want to train \
from the Hugging Face hub"
},
)
dataset_name: Optional[str] = field(
dataset_name: str = field(
default="vietgpt/wikipedia_vi",
metadata={"help": "The instruction dataset to use"},
)
Expand Down

0 comments on commit 16cfa63

Please sign in to comment.