Spaces:

Songyou
/

LLM-fastAPI

Sleeping

LLM-fastAPI / models /transformer /encode_decode /clones.py

add files

f3b11f9 almost 2 years ago

162 Bytes

import copy

import torch.nn as nn


def clones(module, N):
    "Produce N identical layers."
    return nn.ModuleList([copy.deepcopy(module) for _ in range(N)])

	import copy

	import torch.nn as nn


	def clones(module, N):
	"Produce N identical layers."
	return nn.ModuleList([copy.deepcopy(module) for _ in range(N)])