Spaces:

aexyb
/

Diffusion

Runtime error

aexyb commited on Aug 14, 2024

Commit

e2e5f33

verified ·

1 Parent(s): 9584a10

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -261,7 +261,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 #Compile
-    pipe = StableDiffusionXLPipeline.from_pretrained(
         "SG161222/RealVisXL_V4.0_Lightning", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
         torch_dtype=torch.float16,
         use_safetensors=True,

 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 #Compile
+pipe = StableDiffusionXLPipeline.from_pretrained(
         "SG161222/RealVisXL_V4.0_Lightning", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
         torch_dtype=torch.float16,
         use_safetensors=True,