diff --git a/server/text_generation_server/models/custom_modeling/qwen2_vl.py b/server/text_generation_server/models/custom_modeling/qwen2_vl.py index 907fa163..4edd336d 100644 --- a/server/text_generation_server/models/custom_modeling/qwen2_vl.py +++ b/server/text_generation_server/models/custom_modeling/qwen2_vl.py @@ -29,10 +29,7 @@ else: from transformers.activations import ACT2FN import torch.nn.functional as F -from text_generation_server.layers.layernorm import ( - FastLayerNorm, - FastRMSNorm -) +from text_generation_server.layers.layernorm import FastLayerNorm, FastRMSNorm from text_generation_server.layers import ( TensorParallelColumnLinear, TensorParallelRowLinear,