macrdel commited on
Commit
72437e3
·
1 Parent(s): e37bfc9

update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -11,7 +11,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
11
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
12
  model_repo = "black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use
13
 
14
- pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype, token=f"{os.environ.get('tkn')}").to(device)
 
 
 
 
15
  #pipe.enable_model_cpu_offload()
16
 
17
  MAX_SEED = np.iinfo(np.int32).max
 
11
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
12
  model_repo = "black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use
13
 
14
+ pipe = FluxPipeline.from_pretrained(model_repo,
15
+ torch_dtype=torch_dtype,
16
+ token=f"{os.environ.get('tkn')}",
17
+ force_download=True,
18
+ ).to(device)
19
  #pipe.enable_model_cpu_offload()
20
 
21
  MAX_SEED = np.iinfo(np.int32).max