Skip to content

Commit 8267ee5

Browse files
committed
change hidden dim to 128
1 parent 1976d07 commit 8267ee5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxruntime/test/providers/nv_tensorrt_rtx/test_nv_trt_rtx_ep_util.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ void CreateLargeLLMModel(const PathString& model_path, const PathString& externa
277277
// Model parameters (example: 24 layers, 4096 hidden dim, 32 attention heads, 8 kv heads => GQA)
278278
int batch_size = 1;
279279
int num_layers = 1;
280-
int hidden_dim = 512;
280+
int hidden_dim = 128;
281281
int q_num_heads = 8;
282282
int kv_num_heads = 1; // GQA: q_num_heads > kv_num_heads, and divisible.
283283
int seq_length = 128; // Short, for demonstration.

0 commit comments

Comments
 (0)