diff --git a/elastic-ctr/client/demo/demo.cpp b/elastic-ctr/client/demo/demo.cpp index 2553697fb51f41678dab7daa9fb4119c68f66db1..59a094ee94839822e918d61c21334415c7cee606 100644 --- a/elastic-ctr/client/demo/demo.cpp +++ b/elastic-ctr/client/demo/demo.cpp @@ -30,7 +30,7 @@ DEFINE_int32(batch_size, 10, "Infernce batch_size"); DEFINE_string(test_file, "", "test file"); const int VARIABLE_NAME_LEN = 256; -const int CTR_EMBEDDING_TABLE_SIZE = 400000001; +const int CTR_EMBEDDING_TABLE_SIZE = 100000001; struct Sample { std::map> slots; diff --git a/elastic-ctr/client/demo/elastic_ctr.py b/elastic-ctr/client/demo/elastic_ctr.py index abd53ff26a8fea6680942d7359dc06682dd632c2..30873437c67d58c3a816ad344019bbeaa7503d54 100644 --- a/elastic-ctr/client/demo/elastic_ctr.py +++ b/elastic-ctr/client/demo/elastic_ctr.py @@ -22,7 +22,7 @@ from elastic_ctr_api import ElasticCTRAPI BATCH_SIZE = 3 SERVING_IP = "127.0.0.1" SLOT_CONF_FILE = "./conf/slot.conf" -CTR_EMBEDDING_TABLE_SIZE = 400000001 +CTR_EMBEDDING_TABLE_SIZE = 100000001 SLOTS = []