mirror of https://github.com/InternLM/InternLM
[CI]: fix flake8 error
parent
2243fb82aa
commit
4c022ff0e8
|
@ -23,7 +23,7 @@ from typing import Callable, List, Optional
|
|||
import streamlit as st
|
||||
import torch
|
||||
from torch import nn
|
||||
from transformers.generation.utils import (LogitsProcessorList,
|
||||
from transformers.generation.utils import (LogitsProcessorList,
|
||||
StoppingCriteriaList)
|
||||
from transformers.utils import logging
|
||||
|
||||
|
|
Loading…
Reference in New Issue