Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ os.environ["HF_HOME"] = cache_path
|
|
23 |
|
24 |
torch.backends.cuda.matmul.allow_tf32 = True
|
25 |
|
26 |
-
pipe = FluxPipeline.from_pretrained("
|
27 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|
28 |
|
29 |
# Load LoRAs from JSON file
|
|
|
23 |
|
24 |
torch.backends.cuda.matmul.allow_tf32 = True
|
25 |
|
26 |
+
pipe = FluxPipeline.from_pretrained("AlekseyCalvin/PixelwaveFluxSchnell_Diffusers", torch_dtype=torch.bfloat16)
|
27 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|
28 |
|
29 |
# Load LoRAs from JSON file
|