aexyb commited on
Commit
9584a10
·
verified ·
1 Parent(s): 4227bde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -260,7 +260,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
260
 
261
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
262
  #Compile
263
- if torch.cuda.is_available():
264
  pipe = StableDiffusionXLPipeline.from_pretrained(
265
  "SG161222/RealVisXL_V4.0_Lightning", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
266
  torch_dtype=torch.float16,
 
260
 
261
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
262
  #Compile
263
+
264
  pipe = StableDiffusionXLPipeline.from_pretrained(
265
  "SG161222/RealVisXL_V4.0_Lightning", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
266
  torch_dtype=torch.float16,