jbilcke-hf HF staff commited on
Commit
60c8ea5
1 Parent(s): 3841f91

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +6 -4
handler.py CHANGED
@@ -86,9 +86,11 @@ class GenerationConfig:
86
  # The value must be a multiple of 8, plus 1 frame.
87
  # visual glitches appear after about 169 frames, so we don't need more actually
88
  num_frames: int = (8 * 14) + 1
 
 
 
89
 
90
- guidance_scale: float = 5.0
91
- num_inference_steps: int = 30
92
 
93
  # reproducible generation settings
94
  seed: int = -1 # -1 means random seed
@@ -247,10 +249,10 @@ class EndpointHandler:
247
  - width (optional, int, default to 768): width, or horizontal size in pixels.
248
  - height (optional, int, default to 512): height, or vertical size in pixels.
249
  - num_frames (optional, int, default to 129): the numer of frames must be a multiple of 8, plus 1 frame.
250
- - guidance_scale (optional, float, default to 7.5): Guidance scale
251
  - num_inference_steps (optional, int, default to 50): number of inference steps
252
  - seed (optional, int, default to -1): set a random number generator seed, -1 means random seed.
253
- - fps (optional, int, default to 24): FPS of the final video
254
  - double_num_frames (optional, bool): if enabled, the number of frames will be multiplied by 2 using RIFE
255
  - super_resolution (optional, bool): if enabled, the resolution will be multiplied by 2 using Real_ESRGAN
256
  - grain_amount (optional, float): amount of film grain to add to the output video
 
86
  # The value must be a multiple of 8, plus 1 frame.
87
  # visual glitches appear after about 169 frames, so we don't need more actually
88
  num_frames: int = (8 * 14) + 1
89
+
90
+ # values between 3.0 and 4.0 are nice
91
+ guidance_scale: float = 3.5
92
 
93
+ num_inference_steps: int = 50
 
94
 
95
  # reproducible generation settings
96
  seed: int = -1 # -1 means random seed
 
249
  - width (optional, int, default to 768): width, or horizontal size in pixels.
250
  - height (optional, int, default to 512): height, or vertical size in pixels.
251
  - num_frames (optional, int, default to 129): the numer of frames must be a multiple of 8, plus 1 frame.
252
+ - guidance_scale (optional, float, default to 3.5): Guidance scale (values between 3.0 and 4.0 are nice)
253
  - num_inference_steps (optional, int, default to 50): number of inference steps
254
  - seed (optional, int, default to -1): set a random number generator seed, -1 means random seed.
255
+ - fps (optional, int, default to 24): FPS of the final video (eg. 24, 25, 30, 60)
256
  - double_num_frames (optional, bool): if enabled, the number of frames will be multiplied by 2 using RIFE
257
  - super_resolution (optional, bool): if enabled, the resolution will be multiplied by 2 using Real_ESRGAN
258
  - grain_amount (optional, float): amount of film grain to add to the output video