Spaces:
Running
on
Zero
Running
on
Zero
WeichenFan
commited on
Commit
•
f28a5b1
1
Parent(s):
cbd5841
update demo
Browse files- app.py +2 -2
- models/attention.py +1 -1
app.py
CHANGED
@@ -33,8 +33,8 @@ def infer(prompt: str, progress=gr.Progress(track_tqdm=True)):
|
|
33 |
negative_prompt="",
|
34 |
num_inference_steps=50,
|
35 |
guidance_scale=7.5,
|
36 |
-
width=
|
37 |
-
height=
|
38 |
frames=16
|
39 |
)
|
40 |
|
|
|
33 |
negative_prompt="",
|
34 |
num_inference_steps=50,
|
35 |
guidance_scale=7.5,
|
36 |
+
width=768,
|
37 |
+
height=432, #480x288 624x352 432x240 768x432
|
38 |
frames=16
|
39 |
)
|
40 |
|
models/attention.py
CHANGED
@@ -1362,7 +1362,7 @@ class VchitectAttnProcessor:
|
|
1362 |
cross_output = rearrange(cross_output, "B (S T) C -> (B T) S C", T=Frame, B=batchsize)
|
1363 |
cross_output = attn.to_out_context(cross_output)
|
1364 |
|
1365 |
-
hidden_states = hidden_states
|
1366 |
|
1367 |
# Split the attention outputs.
|
1368 |
hidden_states, encoder_hidden_states = (
|
|
|
1362 |
cross_output = rearrange(cross_output, "B (S T) C -> (B T) S C", T=Frame, B=batchsize)
|
1363 |
cross_output = attn.to_out_context(cross_output)
|
1364 |
|
1365 |
+
hidden_states = hidden_states + cross_output
|
1366 |
|
1367 |
# Split the attention outputs.
|
1368 |
hidden_states, encoder_hidden_states = (
|