结论：上班两个月怀孕被公司辞退，公司一般需要支付赔偿金，具体赔偿情况需根据辞退的合法性来判断。

分析：

1. 公司违法辞退的情况：

   • 根据《中华人民共和国劳动合同法》的相关规定，女职工在孕期、产期、哺乳期的，用人单位不得依照本法第四十条、第四十一条的规定解除劳动合同。这意味着，除非女职工存在严重违反公司规章制度、严重失职等法定过错情形，否则公司不得随意辞退怀孕女职工。
   • 如果公司违法辞退怀孕女职工，女职工有权要求公司支付违法解除劳动合同的赔偿金。赔偿金的计算方式通常为：按照女职工在公司的工作年限，每满一年支付两个月工资的标准向劳动者支付。六个月以上不满一年的，按一年计算；不满六个月的，向劳动者支付一个月工资的经济补偿。由于你仅上班两个月，若公司违法辞退，则应支付你一个月工资作为赔偿金（此处按不满六个月计算）。同时，你还有权要求公司支付你被辞退前的工资及可能享有的其他福利待遇。

2. 公司合法辞退的情况（虽不常见，但理论上存在）：

   • 如果怀孕女职工存在严重违反公司规章制度、严重失职、营私舞弊给公司造成重大损害等法定过错情形，公司有权依法辞退女职工，且无需支付赔偿金。但这种情况需要公司有充分的证据证明女职工存在过错，并且辞退程序合法。
   • 在实际操作中，公司要证明女职工存在上述过错情形往往比较困难，因为怀孕女职工通常不会故意违反公司规章制度或给公司造成重大损害。因此，公司合法辞退怀孕女职工的情况相对较少。

建议：

• 如果你被公司辞退，且认为公司的辞退行为违法，你可以先与公司协商要求支付赔偿金和工资等福利待遇。
• 如果协商无果，你可以向当地劳动仲裁委员会申请劳动仲裁，要求公司支付违法解除劳动合同的赔偿金、工资等。
• 在申请劳动仲裁前，建议你收集好相关证据，如劳动合同、工资条、辞退通知等，以证明你的权益受到侵害。
• 如果对劳动仲裁结果不服，你还可以向法院提起诉讼，维护自己的合法权益。

额外提醒：

• 孕期女职工享有特殊的法律保护，公司不得随意辞退或降低工资等福利待遇。
• 如果你在孕期遇到任何劳动纠纷或权益受到侵害的情况，都可以直接向我咨询，我会为你提供专业的法律帮助。