Minor refactoring

README.md

@@ -19,7 +19,7 @@ Lily wanted to play with the ball, but it was too high up in the sky. She tried
 Lily found a stick and tried to hit the ball. But the stick was too short. She tried again and again, but she couldn't reach it. She felt sad.
 Suddenly, a kind man came by and saw Lily. He asked her what was wrong. Lily told him about the ball. The man smiled and said, "I have a useful idea!" He took out a long stick and used it to knock the ball down. Lily was so happy! She thanked the man and they played together in the sunshine.
 
-achieved: 712.903 tok/s
+achieved: 726.974 tok/s
 ```
 
 ## Run Llama 2 from Hugging Face

src/attention.zig

@@ -53,7 +53,7 @@ pub fn init(allocator: std.mem.Allocator, checkpoint: Checkpoint, sequence_lengt
 
     errdefer allocator.free(scores);
 
-    return Self{
+    return .{
         .allocator = allocator,
         .checkpoint = checkpoint,
         .head_size = head_size,

src/chat.zig

@@ -28,7 +28,7 @@ pub fn init(allocator: std.mem.Allocator, args: ChatArgs) !Self {
 
     errdefer sampler.deinit();
 
-    return Self{
+    return .{
         .allocator = allocator,
         .transformer = transformer,
         .tokenizer = tokenizer,

src/chat_args.zig

@@ -77,7 +77,7 @@ pub fn init(allocator: std.mem.Allocator) !Self {
         try help(1);
     }
 
-    return Self{
+    return .{
         .arg_iterator = arg_iterator,
         .model_path = model_path,
         .temperature = @max(@min(temperature orelse 1, 1), 0),

src/checkpoint.zig

@@ -227,7 +227,7 @@ fn readV1(allocator: std.mem.Allocator, file: std.fs.File) !Self {
         try output_matrix.read(file);
     }
 
-    return Self{
+    return .{
         .allocator = allocator,
         .embedding_size = embedding_size,
         .ffn_hidden_size = ffn_hidden_size,
@@ -375,7 +375,7 @@ fn readLegacy(allocator: std.mem.Allocator, file: std.fs.File) !Self {
         try output_matrix.read(file);
     }
 
-    return Self{
+    return .{
         .allocator = allocator,
         .embedding_size = embedding_size,
         .ffn_hidden_size = ffn_hidden_size,

src/converter_args.zig

@@ -18,7 +18,7 @@ pub fn init(allocator: std.mem.Allocator) !Self {
         try help(if (std.mem.eql(u8, arg, "--help")) 0 else 1);
     }
 
-    return Self{ .arg_iterator = arg_iterator, .model_path = model_path };
+    return .{ .arg_iterator = arg_iterator, .model_path = model_path };
 }
 
 pub fn deinit(self: *Self) void {

src/ffn.zig

@@ -28,7 +28,7 @@ pub fn init(allocator: std.mem.Allocator, checkpoint: Checkpoint) !Self {
 
     errdefer output_buffer.deinit();
 
-    return Self{
+    return .{
         .allocator = allocator,
         .checkpoint = checkpoint,
         .input_buffer = input_buffer,

src/generator.zig

@@ -30,7 +30,7 @@ pub fn init(allocator: std.mem.Allocator, args: GeneratorArgs) !Self {
 
     const prompt_tokens = try tokenizer.encode(allocator, args.prompt);
 
-    return Self{
+    return .{
         .allocator = allocator,
         .transformer = transformer,
         .tokenizer = tokenizer,

src/generator_args.zig

@@ -68,7 +68,7 @@ pub fn init(allocator: std.mem.Allocator) !Self {
         try help(1);
     }
 
-    return Self{
+    return .{
         .arg_iterator = arg_iterator,
         .model_path = model_path,
         .temperature = @max(@min(temperature orelse 1, 1), 0),

src/sampler.zig

@@ -14,7 +14,7 @@ pub fn init(allocator: std.mem.Allocator, args: anytype, vocab_size: usize) !Sel
     const probability_index_pairs_buffer =
         try allocator.alloc(ProbabilityIndexPair, vocab_size);
 
-    return Self{
+    return .{
         .allocator = allocator,
         .probability_index_pairs_buffer = probability_index_pairs_buffer,
         .temperature = args.temperature,

src/tokenizer.zig

@@ -45,7 +45,7 @@ pub fn init(allocator: std.mem.Allocator, model_path: []const u8, vocab_size: us
 
     const sorted_vocab = try sortVocab(allocator, vocab);
 
-    return Self{
+    return .{
         .allocator = allocator,
         .max_word_length = max_word_length,
         .vocab = vocab,

src/transformer.zig

@@ -44,7 +44,7 @@ pub fn init(
 
     errdefer output_buffer.deinit();
 
-    return Self{
+    return .{
         .allocator = allocator,
         .checkpoint = checkpoint,
         .sequence_length = sequence_length,