slobers commited on
Commit
3aab0e2
·
verified ·
1 Parent(s): a824051

Update pyproject.toml

Browse files
Files changed (1) hide show
  1. pyproject.toml +7 -1
pyproject.toml CHANGED
@@ -9,10 +9,11 @@ requires-python = ">=3.10,<3.13"
9
  version = "8"
10
  dependencies = [
11
  "diffusers==0.31.0",
 
12
  "transformers==4.46.2",
13
  "accelerate==1.1.0",
14
  "omegaconf==2.3.0",
15
- "torch==2.5.1",
16
  "protobuf==5.28.3",
17
  "sentencepiece==0.2.0",
18
  "torchao==0.6.1",
@@ -25,5 +26,10 @@ dependencies = [
25
  repository = "slobers/Flux.1.Schnella"
26
  revision = "e34d670e44cecbbc90e4962e7aada2ac5ce8b55b"
27
 
 
 
 
 
 
28
  [project.scripts]
29
  start_inference = "main:main"
 
9
  version = "8"
10
  dependencies = [
11
  "diffusers==0.31.0",
12
+ "para-attn==0.3.15",
13
  "transformers==4.46.2",
14
  "accelerate==1.1.0",
15
  "omegaconf==2.3.0",
16
+ "torch==2.6.0",
17
  "protobuf==5.28.3",
18
  "sentencepiece==0.2.0",
19
  "torchao==0.6.1",
 
26
  repository = "slobers/Flux.1.Schnella"
27
  revision = "e34d670e44cecbbc90e4962e7aada2ac5ce8b55b"
28
 
29
+ [[tool.edge-maxxing.models]]
30
+ repository = "black-forest-labs/FLUX.1-schnell"
31
+ revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
32
+ exclude = ["transformer", "text_encoder_2", "text_encoder"]
33
+
34
  [project.scripts]
35
  start_inference = "main:main"