pep 8

rasbt · Jan 12, 2025 · b46f2d0 · b46f2d0
1 parent aa0c7fc
commit b46f2d0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ch05/07_gpt_to_llama/tests/tests.py b/ch05/07_gpt_to_llama/tests/tests.py
@@ -20,8 +20,10 @@
 
 transformers_version = transformers.__version__
 
-# LitGPT code from https://github.com/Lightning-AI/litgpt/blob/main/litgpt/model.py
+# LitGPT code function `litgpt_build_rope_cache` from https://github.com/Lightning-AI/litgpt/blob/main/litgpt/model.py
 # LitGPT is licensed under Apache v2: https://github.com/Lightning-AI/litgpt/blob/main/LICENSE
+
+
 def litgpt_build_rope_cache(
     seq_len: int,
     n_elem: int,