WeichenFan commited on
Commit
f28a5b1
1 Parent(s): cbd5841

update demo

Browse files
Files changed (2) hide show
  1. app.py +2 -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=432,
37
- height=240, #480x288 624x352 432x240 768x432
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*1.1 + cross_output
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 = (