Skip to content
This repository has been archived by the owner on Aug 30, 2024. It is now read-only.

Commit

Permalink
Update cal_diff.py
Browse files Browse the repository at this point in the history
  • Loading branch information
intellinjun authored Jan 24, 2024
1 parent 239d59e commit 99db172
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions scripts/cal_diff.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,10 @@ def cmpData(numa, numb):
args = parser.parse_args()

woq_configs = {
"fp32": {"use_cache":True, "use_quant":False},
# "ggml_int4": {"compute_dtype":"int8", "weight_dtype":"int4", "use_cache":True, "use_ggml":True},
"jblas_int4": {"compute_dtype":"int8", "weight_dtype":"int4", "use_cache":True},
# "jblas_int8": {"compute_dtype":"bf16", "weight_dtype":"int8", "use_cache":True},
"fp32": {"use_quant":False},
# "ggml_int4": {"compute_dtype":"int8", "weight_dtype":"int4", "use_ggml":True},
"jblas_int4": {"compute_dtype":"int8", "weight_dtype":"int4"},
# "jblas_int8": {"compute_dtype":"bf16", "weight_dtype":"int8"},

}
prompt = "What is the meaning of life?"
Expand Down

0 comments on commit 99db172

Please sign in to comment.