Skip to content

GitLab

Commit 9060b980 authored by Casper Hansen's avatar Casper Hansen
Browse files

Remove imports

parent 37db5759
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
awq/modules/fused/model.py
View file @ 9060b980
import torch import torch
import torch.nn as nn import torch.nn as nn
from awq.modules.fused.block import MPTBlock, FalconDecoderLayer, GptBigCodeBlock from awq.modules.fused.block import MPTBlock, FalconDecoderLayer
from transformers.modeling_outputs import BaseModelOutputWithPast, BaseModelOutputWithPastAndCrossAttentions from transformers.modeling_outputs import BaseModelOutputWithPast
class MPTModel(nn.Module): class MPTModel(nn.Module):
def __init__(self, vocab_size, blocks, wte, norm_f): def __init__(self, vocab_size, blocks, wte, norm_f):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment