[BAGEL] [Feature]: Add thinking mode in Bagel multi-stage serving (#2447)
This commit is contained in:
7 files changed
+443
-46
No files matched your search
@@ -326,11 +326,18 @@ class BagelPipeline(nn.Module, DiffusionPipelineProfilerMixin):
|
||||
cfg_text_scale = extra_args.get("cfg_text_scale", 4.0)
|
||||
cfg_img_scale = extra_args.get("cfg_img_scale", 1.5)
|
||||
|
||||
cfg_interval = extra_args.get("cfg_interval", (0.4, 1.0))
|
||||
cfg_renorm_type = extra_args.get("cfg_renorm_type", "global")
|
||||
cfg_renorm_min = extra_args.get("cfg_renorm_min", 0.0)
|
||||
|
||||
gen_params = BagelGenParams(
|
||||
num_timesteps=int(req.sampling_params.num_inference_steps or 50),
|
||||
timestep_shift=3.0,
|
||||
cfg_text_scale=cfg_text_scale,
|
||||
cfg_img_scale=cfg_img_scale,
|
||||
cfg_interval=cfg_interval,
|
||||
cfg_renorm_type=cfg_renorm_type,
|
||||
cfg_renorm_min=cfg_renorm_min,
|
||||
)
|
||||
|
||||
gen_context = {
|
||||
|
||||
Reference in new issue
Block a user