RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Dec 31, 2024

Commit

17b8b1d

verified ·

1 Parent(s): 56daa8c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = 0
 BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 style_list = [
     {

 ENABLE_CPU_OFFLOAD = 0
 BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
+device = torch.device("cuda:0")
 style_list = [
     {