macrdel commited on
Commit
bbdd646
·
1 Parent(s): 19ae9cd

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
11
- model_repo = "models/black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use
12
 
13
  pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
14
  #pipe.enable_model_cpu_offload()
 
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
11
+ model_repo = "black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use
12
 
13
  pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
14
  #pipe.enable_model_cpu_offload()