Yaron Koresh commited on
Commit
e453455
·
verified ·
1 Parent(s): 651ba0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -42,12 +42,12 @@ root.addHandler(handler)
42
 
43
  if torch.cuda.is_available():
44
  device = "cuda"
45
- dtype = torch.float32
46
  else:
47
  device = "cpu"
48
- dtype = torch.float32
49
 
50
- base = "black-forest-labs/FLUX.1-dev"
51
 
52
  # variable data
53
 
@@ -59,7 +59,7 @@ seq=1024
59
  fps=20
60
  width=1280
61
  height=720
62
- image_steps=50
63
  video_steps=30
64
  accu=4.5
65
 
 
42
 
43
  if torch.cuda.is_available():
44
  device = "cuda"
45
+ dtype = torch.bfloat16
46
  else:
47
  device = "cpu"
48
+ dtype = torch.bfloat16
49
 
50
+ base = "black-forest-labs/FLUX.1-schnell"
51
 
52
  # variable data
53
 
 
59
  fps=20
60
  width=1280
61
  height=720
62
+ image_steps=8
63
  video_steps=30
64
  accu=4.5
65