Use AWS Neuron sdk 2.20

[dacorvo]

What does this PR do?

This updates optimum-neuron and NeuronX TGI to use AWS Neuron SDK 2.20.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[JingyaHuang]

For the raised error, @dangdana you could remove the class WeightSeparatedDataParallel since the DP for non-inlined shall have been supported directly from 2.20:

optimum-neuron/optimum/neuron/modeling_diffusion.py

Lines 355 to 361 in 3b381c0

	if to_neuron:
	if Version(__sdk_version__) >= Version("2.20.0"):
	raise NameError(
	"`WeightSeparatedDataParallel` class should be deprecated when neuron sdk 2.20 is out. Please replace it with `torch_neuronx.DataParallel`."
	)
	dp_cls = WeightSeparatedDataParallel
	else:

[dacorvo]

For the raised error, @dangdana you could remove the class WeightSeparatedDataParallel since the DP for non-inlined shall have been supported directly from 2.20:

optimum-neuron/optimum/neuron/modeling_diffusion.py

Lines 355 to 361 in 3b381c0

	if to_neuron:
	if Version(__sdk_version__) >= Version("2.20.0"):
	raise NameError(
	"`WeightSeparatedDataParallel` class should be deprecated when neuron sdk 2.20 is out. Please replace it with `torch_neuronx.DataParallel`."
	)
	dp_cls = WeightSeparatedDataParallel
	else:

Yes, I already fixed it (fyi you pinged someone else ;-) ).

[JingyaHuang]

LGTM, thanks for working on this @dacorvo !

[michaelbenayoun]

LGTM, thanks @dacorvo !