cocktailpeanut commited on
Commit
8de121e
1 Parent(s): 40ffbf4
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ device = devicetorch.get(torch)
10
  if device == "cuda":
11
  dtype = torch.bfloat16
12
  elif device == "mps":
13
- dtype = torch.bfloat16
14
  else:
15
  dtype = torch.float32
16
  #dtype = torch.bfloat16
 
10
  if device == "cuda":
11
  dtype = torch.bfloat16
12
  elif device == "mps":
13
+ dtype = torch.float16
14
  else:
15
  dtype = torch.float32
16
  #dtype = torch.bfloat16
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  accelerate
2
  #git+https://github.com/huggingface/diffusers.git@flux-pipeline
3
- git+https://github.com/huggingface/diffusers.git
4
  invisible_watermark
5
  #torch
6
  transformers==4.42.4
 
1
  accelerate
2
  #git+https://github.com/huggingface/diffusers.git@flux-pipeline
3
+ git+https://github.com/peanutcocktail/diffusers.git
4
  invisible_watermark
5
  #torch
6
  transformers==4.42.4