From a49ab6e656488b3c65c945390a722650deffede9 Mon Sep 17 00:00:00 2001 From: anirudTT Date: Mon, 8 Apr 2024 18:46:06 +0000 Subject: [PATCH] =?UTF-8?q?***=20AUTOMATED=20COMMIT=20|=20Applied=20Code?= =?UTF-8?q?=20Formatting=20and=20Cleanup=20=E2=9C=A8=20=F0=9F=8D=B0=20?= =?UTF-8?q?=E2=9C=A8***?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py b/model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py index 6a9675e9..1c240ece 100644 --- a/model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py +++ b/model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py @@ -2,11 +2,11 @@ import os - +import pybuda from pybuda._C.backend_api import BackendDevice from transformers import AlbertForMaskedLM, AlbertTokenizer -import pybuda + def run_albert_masked_lm_pytorch(size="base", variant="v2"): available_devices = pybuda.detect_available_devices()