Update main.py

Xilinx · Dec 13, 2024 · 88e81a8 · 88e81a8
1 parent 9596911
commit 88e81a8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py
@@ -374,6 +374,7 @@ def main(args):
 
     if args.bias_corr:
         model = add_zero_bias_to_linear(model)
+
     model = offload_model(model)
 
     with torch.no_grad():