[frontend] Output bert params (#249)

buddy-compiler · Dec 9, 2023 · f01d596 · f01d596
1 parent 3322a4f
commit f01d596
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)