revert changes in examples

awgu · May 21, 2024 · 95d36c2 · 95d36c2
1 parent e2c027f
commit 95d36c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/example_chat_completion.py b/example_chat_completion.py
@@ -1,8 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # This software may be used and distributed in accordance with the terms of the Llama 3 Community License Agreement.
 
-import os
-
 from typing import List, Optional
 
 import fire
@@ -30,10 +28,6 @@ def main(
 
  `max_gen_len` is optional because finetuned models are able to stop generations naturally.
  """
- assert 1 <= max_seq_len <= 8192, f"max_seq_len must be between 1 and 8192, got {max_seq_len}."
- assert os.path.isdir(ckpt_dir), f"Checkpoint directory '{ckpt_dir}' does not exist."
- assert os.path.isfile(tokenizer_path), f"Tokenizer file '{tokenizer_path}' does not exist."
-
  generator = Llama.build(
  ckpt_dir=ckpt_dir,
  tokenizer_path=tokenizer_path,
@@ -87,4 +81,4 @@ def main(
 
 
 if __name__ == "__main__":
- fire.Fire(main)
+ fire.Fire(main)
diff --git a/example_text_completion.py b/example_text_completion.py
@@ -1,8 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # This software may be used and distributed in accordance with the terms of the Llama 3 Community License Agreement.
 
-import os
-
 from typing import List
 
 import fire
@@ -26,10 +24,6 @@ def main(
  The context window of llama3 models is 8192 tokens, so `max_seq_len` needs to be <= 8192.
  `max_gen_len` is needed because pre-trained models usually do not stop completions naturally.
  """
- assert 1 <= max_seq_len <= 8192, f"max_seq_len must be between 1 and 8192, got {max_seq_len}."
- assert os.path.isdir(ckpt_dir), f"Checkpoint directory '{ckpt_dir}' does not exist."
- assert os.path.isfile(tokenizer_path), f"Tokenizer file '{tokenizer_path}' does not exist."
-
  generator = Llama.build(
  ckpt_dir=ckpt_dir,
  tokenizer_path=tokenizer_path,
@@ -67,4 +61,4 @@ def main(
 
 
 if __name__ == "__main__":
- fire.Fire(main)
+ fire.Fire(main)