FluxBooru-CFG3.5

Running on Zero

bghira commited on about 14 hours ago

Commit

f11b8e2

•

1 Parent(s): aa8ba97

update to fluxbooru v0.2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from custom_pipeline import FluxPipeline
 import gradio as gr
 # Load the pre-trained diffusion model
-pipe = FluxPipeline.from_pretrained('terminusresearch/flux-booru-cfg3.5', torch_dtype=torch.bfloat16)
 pipe.to('cuda')
 import re

 import gradio as gr
 # Load the pre-trained diffusion model
+pipe = FluxPipeline.from_pretrained('terminusresearch/flux-booru-v0.2', torch_dtype=torch.bfloat16)
 pipe.to('cuda')
 import re