From dd44afda39fce1a16e7b2ecf3bb3ea727756a5b8 Mon Sep 17 00:00:00 2001 From: Rex Cheng Date: Sun, 22 Dec 2024 11:16:38 -0600 Subject: [PATCH] Enforce quote consistency --- gradio_demo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gradio_demo.py b/gradio_demo.py index d2aa68d..9c01b0c 100644 --- a/gradio_demo.py +++ b/gradio_demo.py @@ -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