Hugging Face's logo

Spaces:
jiuface
/
flux-dev-multi-lora
Running on Zero

App Files Community
jiuface commited on
Commit
be3d458
·
verified ·
1 Parent(s): da3a145

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,8 +33,8 @@ dtype = torch.bfloat16
33
  device = "cuda:0"
34
 
35
  print(device)
36
- base_model = "black-forest-labs/FLUX.1-dev"
37
-
38
  # load pipe
39
 
40
  txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
 
33
  device = "cuda:0"
34
 
35
  print(device)
36
+ #base_model = "black-forest-labs/FLUX.1-dev"
37
+ base_model = "black-forest-labs/FLUX.1-Krea-dev"
38
  # load pipe
39
 
40
  txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)