Spaces:

mgbam
/

image

Runtime error

mgbam commited on Jan 29

Commit

ed1226e

verified ·

1 Parent(s): 871d5f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from diffusers import AutoencoderKL
 import numpy as np
 import gradio as gr
-# Configure device and disable FlashAttention
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.bfloat16 if device == "cuda" else torch.float32
 print(f"Using device: {device}")

 import numpy as np
 import gradio as gr
+# Configure device
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.bfloat16 if device == "cuda" else torch.float32
 print(f"Using device: {device}")