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 streamlit as st
|
||||||
import torch
|
import torch
|
||||||
from torch import nn
|
from torch import nn
|
||||||
from transformers.generation.utils import (LogitsProcessorList,
|
from transformers.generation.utils import (LogitsProcessorList,
|
||||||
StoppingCriteriaList)
|
StoppingCriteriaList)
|
||||||
from transformers.utils import logging
|
from transformers.utils import logging
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue