Skip to content

Commit

Permalink
Enforce quote consistency
Browse files Browse the repository at this point in the history
  • Loading branch information
hkchengrex committed Dec 22, 2024
1 parent 5f0e921 commit dd44afd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions gradio_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,11 @@

log = logging.getLogger()

device = "cpu"
device = 'cpu'
if torch.backends.mps.is_available():
device = "mps"
device = 'mps'
elif torch.cuda.is_available():
device = "cuda"
device = 'cuda'

dtype = torch.bfloat16

Expand Down

0 comments on commit dd44afd

Please sign in to comment.