images

Runtime error

aimersion commited on Aug 27, 2024

Commit

2cb4198

verified ·

1 Parent(s): 641702f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load the diffusion pipeline from the gated repository
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048

 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load the diffusion pipeline from the gated repository
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell")
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048