mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-07-04 07:26:36 +00:00
17 lines
597 B
Python
17 lines
597 B
Python
|
# Copyright (c) Sebastian Raschka under Apache License 2.0 (see LICENSE.txt).
|
||
|
|
# Source for "Build a Large Language Model From Scratch"
|
||
|
|
# - https://www.manning.com/books/build-a-large-language-model-from-scratch
|
||
|
|
# Code: https://github.com/rasbt/LLMs-from-scratch
|
||
|
|
|
||
|
|
# File for internal use (unit tests)
|
||
|
|
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
|
||
|
|
def test_gpt_class_finetune():
|
||
|
|
command = ["python", "ch06/01_main-chapter-code/gpt_class_finetune.py", "--test_mode"]
|
||
|
|
|
||
|
|
result = subprocess.run(command, capture_output=True, text=True)
|
||
|
|
assert result.returncode == 0, f"Script exited with errors: {result.stderr}"
|