Setting llama globals as uninitialized

Author: IanNod

python/turbine_models/custom_models/stateless_llama.py

@@ -125,8 +125,10 @@ class StateUpdateModule(CompiledModule):
             )
         else:
             params = export_parameters(mod)
-        global_state = export_global(abstractify(global_pkv), mutable=True)
-        global_seq_step = export_global(AbstractIndex, mutable=True)
+        global_state = export_global(
+            abstractify(global_pkv), uninitialized=True, mutable=True
+        )
+        global_seq_step = export_global(AbstractIndex, uninitialized=True, mutable=True)
 
         def run_initialize(self, x=AbstractTensor(BATCH_SIZE, None, dtype=torch.int64)):
             init_const = [x.dynamic_dim(1) < MAX_STEP_SEQ]