FLUX.1-schnell-smashed

Running on Zero

davidberenstein1957 commited on 6 days ago

Commit

080bc90

verified ·

1 Parent(s): a8b3bdd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "CPU"
 pipe = PrunaModel.from_hub(
-    "PrunaAI/Segmind-Vega-smashed"
 )
 pipe.move_to_device(device)

 device = "cuda" if torch.cuda.is_available() else "CPU"
 pipe = PrunaModel.from_hub(
+    "PrunaAI/FLUX.1-schnell-smashed-test"
 )
 pipe.move_to_device(device)