LibreFLUX_LoRAs_Gallery

Runtime error

AlekseyCalvin commited on Oct 9, 2024

Commit

58b2d1a

verified ·

1 Parent(s): c78df80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ os.environ["HF_HOME"] = cache_path
 torch.backends.cuda.matmul.allow_tf32 = True
-dtype = torch.float16
 pipe = FluxWithCFGPipeline.from_pretrained(
     "ostris/OpenFLUX.1", torch_dtype=dtype
 ).to("cuda")

 torch.backends.cuda.matmul.allow_tf32 = True
+dtype = torch.bfloat16
 pipe = FluxWithCFGPipeline.from_pretrained(
     "ostris/OpenFLUX.1", torch_dtype=dtype
 ).to("cuda")