Skip to content

Commit

Permalink
remove unused code
Browse files Browse the repository at this point in the history
  • Loading branch information
rasbt committed Oct 14, 2024
1 parent 1d0c00a commit 1f60feb
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions ch05/08_memory_efficient_weight_loading/previous_chapters.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,20 +6,15 @@
# This file collects all the relevant code that we covered thus far
# throughout Chapters 2-5.

import json
import os
import urllib

import numpy as np
import tensorflow as tf
import torch
import torch.nn as nn
from tqdm import tqdm


#####################################
# Chapter 3
#####################################


class MultiHeadAttention(nn.Module):
def __init__(self, d_in, d_out, context_length, dropout, num_heads, qkv_bias=False):
super().__init__()
Expand Down

0 comments on commit 1f60feb

Please sign in to comment.