Spaces:

ISEEKYAN
/

megatron_memory_estimator_old

Running

ISEEKYAN commited on 19 days ago

Commit

04b038b

verified ·

1 Parent(s): 2a41314

fix bug of CP validate

Files changed (1) hide show

webui/script.js CHANGED Viewed

@@ -253,12 +253,12 @@ async function submitForm() {
         errors.push(`TP*PP*CP (${tp * pp * cp}) > GPUs (${num_gpus}).`);
     }
     if (etp){
-        if (etp * pp * cp * ep > num_gpus) {
-            errors.push(`ETP*PP*CP*EP (${etp * pp * cp * ep}) > GPUs (${num_gpus}).`);
         }
     } else {
-        if (tp * pp * cp * ep > num_gpus) {
-            errors.push(`TP*PP*CP*EP (${tp * pp * cp * ep}) > GPUs (${num_gpus}) when ETP is not set.`);
         }
     }
     if (num_kv_heads && tp > num_kv_heads) {

         errors.push(`TP*PP*CP (${tp * pp * cp}) > GPUs (${num_gpus}).`);
     }
     if (etp){
+        if (etp * pp * ep > num_gpus) {
+            errors.push(`ETP*PP*EP (${etp * pp * ep}) > GPUs (${num_gpus}).`);
         }
     } else {
+        if (tp * pp * ep > num_gpus) {
+            errors.push(`TP*PP*EP (${tp * pp * ep}) > GPUs (${num_gpus}) when ETP is not set.`);
         }
     }
     if (num_kv_heads && tp > num_kv_heads) {