Skip to content
This repository has been archived by the owner on Oct 16, 2023. It is now read-only.

update palm by combining gemini #49

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

ZijianYY
Copy link

No description provided.

@@ -148,9 +150,11 @@ def _forward(self, x):

# aggregate values
if self.multi_query:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

那不需要if判断了,两个分支逻辑一模一样

tflop = 0
default_pg = ProcessGroup(tp_degree=gpc.config.TPDEGREE)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

不要使用gpc.config了。它是老接口的配置文件。tp_degree可以通过args设置。

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants