ovi054 commited on
Commit
1011e1d
·
verified ·
1 Parent(s): e5debfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -5,8 +5,6 @@ import spaces
5
  import torch
6
  from diffusers import QwenImagePipeline
7
 
8
- from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
9
-
10
  dtype = torch.bfloat16
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
 
@@ -33,11 +31,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
33
  pipe.unload_lora_weights()
34
  pipe.load_lora_weights(lora_id.strip())
35
 
36
- apply_cache_on_pipe(
37
- pipe,
38
- # residual_diff_threshold=0.2,
39
- )
40
-
41
  try:
42
  image = pipe(
43
  prompt=prompt,
 
5
  import torch
6
  from diffusers import QwenImagePipeline
7
 
 
 
8
  dtype = torch.bfloat16
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
 
 
31
  pipe.unload_lora_weights()
32
  pipe.load_lora_weights(lora_id.strip())
33
 
34
+
 
 
 
 
35
  try:
36
  image = pipe(
37
  prompt=prompt,