Merge pull request #123 from aws-neuron/fazhangm/fix/hf_pretrained_clip_large_inference_on_inf2

fix: Fix the error tuple' object has no attribute 'pooler_output' in hf_pretrained_clip_large_inference_on_inf2.ipynb

Author: fazhangm

torch-neuronx/inference/hf_pretrained_clip_large_inference_on_inf2.ipynb

@@ -94,7 +94,7 @@
                 "\n",
                 "# Create the input preprocessor and model\n",
                 "processor = CLIPProcessor.from_pretrained(model_name)\n",
-                "model = CLIPModel.from_pretrained(model_name, return_dict=False)\n",
+                "model = CLIPModel.from_pretrained(model_name, return_dict=True)\n",
                 "model.eval()\n",
                 "\n",
                 "# Get text captions for the model to classify the image against\n",
@@ -146,7 +146,7 @@
                 "output_neuron = model_neuron(*example)\n",
                 "\n",
                 "# Compare the results\n",
-                "cpu_top5 = output_cpu[0][0].softmax(dim=-1).topk(5)\n",
+                "cpu_top5 = output_cpu.logits_per_image[0].softmax(dim=-1).topk(5)\n",
                 "neuron_top5 = output_neuron[0][0].softmax(dim=-1).topk(5)\n",
                 "\n",
                 "print('CPU top 5 classifications')\n",