Update app.py

app.py

@@ -11,7 +11,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "CPU"
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", dtype=dtype).to("cpu")
-smash_config = SmashConfig()
+smash_config = SmashConfig(device="cpu")
 smash_config["factorizer"] = "qkv_diffusers"
 smash_config["cacher"] = "fora"
 smash_config["fora_interval"] = 2