mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-08-31 12:00:23 +00:00
style checks
This commit is contained in:
parent
d311bae25a
commit
51f4980a42
@ -254,7 +254,7 @@ if __name__ == "__main__":
|
|||||||
help=(
|
help=(
|
||||||
"Whether to use 'pretrained' or 'random' weights."
|
"Whether to use 'pretrained' or 'random' weights."
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--trainable_layers",
|
"--trainable_layers",
|
||||||
type=str,
|
type=str,
|
||||||
@ -290,7 +290,6 @@ if __name__ == "__main__":
|
|||||||
else:
|
else:
|
||||||
raise ValueError("Invalid --trainable_token argument")
|
raise ValueError("Invalid --trainable_token argument")
|
||||||
|
|
||||||
|
|
||||||
###############################
|
###############################
|
||||||
# Load model
|
# Load model
|
||||||
###############################
|
###############################
|
||||||
|
Loading…
x
Reference in New Issue
Block a user