mirror of https://github.com/InternLM/InternLM
fix lint
parent
19f8eb4534
commit
86c5da6dd5
|
@ -17,7 +17,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 import AutoTokenizer, AutoModelForCausalLM # isort: skip
|
from transformers import AutoTokenizer, AutoModelForCausalLM # isort: skip
|
||||||
from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList
|
from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList
|
||||||
from transformers.utils import logging
|
from transformers.utils import logging
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue