Silu op opt #4718
Silu op opt #4718
Conversation
|
Thanks for your contribution! |
|
|
| text_encoder_runtime = create_ort_runtime( | ||
| args.model_dir, args.text_encoder_model_prefix, args.model_format, device_id=device_id | ||
| text_encoder_runtime = create_paddle_inference_runtime( | ||
| args.model_dir, |
| resnet_act_fn=resnet_act_fn, | ||
| resnet_groups=resnet_groups, | ||
| resnet_time_scale_shift=resnet_time_scale_shift, | ||
| resnet_pre_temb_non_linearity=False, |
|
|
||
| # 3. down | ||
| down_block_res_samples = (sample,) | ||
| down_nonlinear_temb = self.down_resnet_temb_nonlinearity(emb) |
There was a problem hiding this comment.
这应该给个判断吧，只有resnet_pre_temb_non_linearity=True的时候才做这个
|
|
||
| # 3. down | ||
| down_block_res_samples = (sample,) | ||
| down_nonlinear_temb = self.down_resnet_temb_nonlinearity(emb) |
There was a problem hiding this comment.
考虑这里是否需要对resnet_pre_temb_non_linearity先进行判断后再对down_nonlinear_temb进行赋值
like
if resnet_pre_temb_non_linearity:
down_nonlinear_temb = self.down_resnet_temb_nonlinearity(emb)
else:
down_nonlinear_temb = emb
|
This Pull Request is stale because it has been open for 60 days with no activity. 当前Pull Request 60天内无活动，被标记为stale。 |
|
This Pull Request is stale because it has been open for 60 days with no activity. 当前Pull Request 60天内无活动，被标记为stale。 |
|
This Pull Request is stale because it has been open for 60 days with no activity. 当前Pull Request 60天内无活动，被标记为stale。 |
|
This Pull Request is stale because it has been open for 60 days with no activity. 当前Pull Request 60天内无活动，被标记为stale。 |
|
Automatically closed by Paddle-bot. |
6 participants
PR types
Performance optimization
PR changes
Models
Description
merge silu op as producer
here is difference: