Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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,
|