Skip to content

Commit

Permalink
*** AUTOMATED COMMIT | Applied Code Formatting and Cleanup ✨ 🍰 ✨***
Browse files Browse the repository at this point in the history
  • Loading branch information
anirudTT committed Apr 8, 2024
1 parent 1b4126e commit a49ab6e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions model_demos/nlp_demos/albert/pytorch_albert_masked_lm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down

0 comments on commit a49ab6e

Please sign in to comment.