jadechoghari commited on
Commit
0455d1d
β€’
1 Parent(s): 144a488

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -17,14 +17,18 @@ pipeline_optimized = FluxPipeline.from_pretrained(
17
  "camenduru/FLUX.1-dev-diffusers",
18
  torch_dtype=torch.bfloat16
19
  ).to("cuda")
20
-
 
 
 
 
21
  pipeline_optimized.transformer.to(memory_format=torch.channels_last)
22
  pipeline_optimized.transformer = torch.compile(
23
  pipeline_optimized.transformer,
24
  mode="max-autotune",
25
  fullgraph=True
26
  )
27
- print("all good")
28
 
29
  pipeline_optimized.transformer = autoquant(
30
  pipeline_optimized.transformer,
 
17
  "camenduru/FLUX.1-dev-diffusers",
18
  torch_dtype=torch.bfloat16
19
  ).to("cuda")
20
+ pipeline_optimized.transformer = autoquant(
21
+ pipeline_optimized.transformer,
22
+ error_on_unseen=False
23
+ )
24
+ print("all good")
25
  pipeline_optimized.transformer.to(memory_format=torch.channels_last)
26
  pipeline_optimized.transformer = torch.compile(
27
  pipeline_optimized.transformer,
28
  mode="max-autotune",
29
  fullgraph=True
30
  )
31
+
32
 
33
  pipeline_optimized.transformer = autoquant(
34
  pipeline_optimized.transformer,