Update configs.py

neuralmagic · Sep 13, 2023 · b6bfd34 · b6bfd34
1 parent 77a671b
commit b6bfd34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/exporters/transforms/kv_cache/configs.py b/src/sparseml/exporters/transforms/kv_cache/configs.py
@@ -125,7 +125,7 @@ class Config:
     multiply_batch_by_num_att_heads=False,
 )
 
-# reusing the transforms for codegen, because it happens to match what we need for gpt neo
+# Reusing the CodeGen transforms because it happens to match what we need for GPTNeo
 additional_transforms_gpt_neo = AdditionalTransformsCodeGen
 
 GPT_NEO_CONFIG = KeyValueCacheConfig(