inoculatemedia commited on
Commit
b986462
·
verified ·
1 Parent(s): 868afbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -12,7 +12,7 @@ import safetensors.torch as sf
12
  import numpy as np
13
  import math
14
  import spaces
15
-
16
  from PIL import Image
17
  from diffusers import AutoencoderKLHunyuanVideo
18
  from transformers import LlamaModel, CLIPTextModel, LlamaTokenizerFast, CLIPTokenizer
@@ -415,12 +415,12 @@ def process(input_image, prompt,
415
  def end_process():
416
  stream.input_queue.push('end')
417
 
418
- def quick_prompts():
419
- quick_prompts = [
420
- 'The girl dances gracefully, with clear movements, full of charm.',
421
- 'A character doing some simple body movements.',
422
- ]
423
- quick_prompts = [[x] for x in quick_prompts]
424
 
425
 
426
  css = make_progress_bar_css()
 
12
  import numpy as np
13
  import math
14
  import spaces
15
+ import flash-attn
16
  from PIL import Image
17
  from diffusers import AutoencoderKLHunyuanVideo
18
  from transformers import LlamaModel, CLIPTextModel, LlamaTokenizerFast, CLIPTokenizer
 
415
  def end_process():
416
  stream.input_queue.push('end')
417
 
418
+
419
+ quick_prompts = [
420
+ 'The girl dances gracefully, with clear movements, full of charm.',
421
+ 'A character doing some simple body movements.',
422
+ ]
423
+ quick_prompts = [[x] for x in quick_prompts]
424
 
425
 
426
  css = make_progress_bar_css()