modification

flexflow · Oct 29, 2024 · 2502c9b · 2502c9b
1 parent 48570a3
commit 2502c9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/flexflow/model.h b/include/flexflow/model.h
@@ -713,7 +713,7 @@ class FFModel {
       const Tensor input,
       int embed_dim,
       int num_heads,
-                                      int num_hidden_layers,
+      int num_hidden_layers,
       int kdim = 0,
       int vdim = 0,
       float dropout = 0.0f,