Update app.py

app.py

@@ -20,7 +20,7 @@ hf_hub_download(repo_id="yucornetto/RAR", filename="rar_xl.bin", local_dir="./")
 
 # @spaces.GPU
 def load_model():
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    device = "cuda" # if torch.cuda.is_available() else "cpu"
     # load config
     rar_model_size = "rar_xl"
     config = demo_util.get_config("configs/training/generator/rar.yaml")
@@ -46,7 +46,7 @@ tokenizer, generator = load_model()
 def demo_infer(
                guidance_scale, randomize_temperature, guidance_scale_pow,
                class_label, seed):
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    device = "cuda" # if torch.cuda.is_available() else "cpu"
     n = 4
     class_labels = [class_label for _ in range(n)]
     torch.manual_seed(seed)