Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
8de121e
1
Parent(s):
40ffbf4
update
Browse files- app.py +1 -1
- 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.
|
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/
|
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
|