Skip to content

Commit c5610a8

Browse files
committed
lint
1 parent 7d99123 commit c5610a8

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

jetstream_pt/cli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ def interactive():
208208

209209

210210
def main():
211+
"""Main function."""
211212
def main_real(argv):
212213
"""Entry point"""
213214
if len(argv) < 2:
@@ -223,6 +224,7 @@ def main_real(argv):
223224
print(
224225
"Invalid arguments. please specify 'list', 'serve', or 'interactive'."
225226
)
227+
226228
app.run(main_real)
227229
return 0
228230

jetstream_pt/fetch_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class ModelInfo:
4040
num_layers: int
4141
# number of kv heads
4242
num_kv_heads: int
43-
43+
4444
head_dim: int
4545
n_reps: int # repeatition for GQA
4646

jetstream_pt/third_party/llama/model_exportable.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,5 +385,5 @@ def transform(val, n_heads):
385385
value, self.params.n_kv_heads or self.params.n_heads
386386
)
387387
res = super().convert_hf_weights(updated)
388-
res['freqs_cis'] = self.freqs_cis
388+
res["freqs_cis"] = self.freqs_cis
389389
return res

0 commit comments

Comments
 (0)