chore: enable input compression in llm use case

zama-ai · Apr 26, 2024 · 19d62c8 · 19d62c8
1 parent 9726072
commit 19d62c8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/use_case_examples/llm/qgpt2_class.py b/use_case_examples/llm/qgpt2_class.py
@@ -183,7 +183,9 @@ def compile(self, configuration: Optional[Configuration] = None) -> Circuit:
         compiler = fhe.Compiler(self.run_numpy, {"q_inputs": "encrypted"})
 
         # Compile the circuit on the calibration quantized data
-        self.circuit = compiler.compile(inputset, configuration=configuration)
+        self.circuit = compiler.compile(
+            inputset, configuration=configuration, compress_input_ciphertexts=True
+        )
 
         # Print the maximum bit-width reached in the circuit
         print(