Spaces:

Artples
/

Playground-V2.5-MultipleImageGeneration

Runtime error

Artples commited on Mar 8

Commit

0ddb132

•

1 Parent(s): 0174b1c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-NUM_IMAGES_PER_PROMPT = 8
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(

 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+NUM_IMAGES_PER_PROMPT = 4
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(