[frontend] output bert params

buddy-compiler · Nov 28, 2023 · abbc6fc · abbc6fc
1 parent cbdfc44
commit abbc6fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/BuddyLlama/import-llama2.py b/examples/BuddyLlama/import-llama2.py
@@ -50,7 +50,7 @@
 # Import the model into MLIR module and parameters.
 with torch.no_grad():
     gm, params = dynamo_compiler.importer(
-        model, torch.tensor([[1 for i in range(40)]], dtype=torch.int64)
+        model, torch.tensor([[1 for _ in range(40)]], dtype=torch.int64)
     )
 
 path_prefix = os.path.dirname(os.path.abspath(__file__))

diff --git a/examples/BuddyPython/bert.py b/examples/BuddyPython/bert.py
@@ -17,4 +17,4 @@
 with torch.no_grad():
     module, params = dynamo_compiler.importer(model, **encoded_text)
     print(module)
-    # print(params)
+    print(params)