fix ChatGLM2 ppo #527 #528

This commit is contained in:
hiyouga committed 2023-08-18 00:34:59 +08:00
1 parent be21fc83f9
commit 9f4c2adc9a
6 files changed
+72 -11

No files matched your search

+1 -5
View File
@@ -1,10 +1,6 @@
import torch
from typing import TYPE_CHECKING, List, Optional, Tuple
from transformers import (
LogitsProcessor,
InfNanRemoveLogitsProcessor,
LogitsProcessorList
)
from transformers import InfNanRemoveLogitsProcessor, LogitsProcessorList
from llmtuner.extras.constants import LAYERNORM_NAMES