Update aoti.py

aoti.py

@@ -32,4 +32,4 @@ def aoti_blocks_load(module: torch.nn.Module, repo_id: str, variant: str | None
                 block_ = _shallow_clone_module(block)
                 unwrap_tensor_subclass_parameters(block_)
                 weights = ZeroGPUWeights(block_.state_dict())
-                block.forward = ZeroGPUCompiledModel(aoti_file, weights)
+                block.forward = ZeroGPUCompiledModel(aoti_file, weights)