Method init

diffsynth/models/sd_motion.py:95–142 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

93
94	class SDMotionModel(torch.nn.Module):
95	def __init__(self):
96	super().__init__()
97	self.motion_modules = torch.nn.ModuleList([
98	TemporalBlock(8, 40, 320, eps=1e-6),
99	TemporalBlock(8, 40, 320, eps=1e-6),
100	TemporalBlock(8, 80, 640, eps=1e-6),
101	TemporalBlock(8, 80, 640, eps=1e-6),
102	TemporalBlock(8, 160, 1280, eps=1e-6),
103	TemporalBlock(8, 160, 1280, eps=1e-6),
104	TemporalBlock(8, 160, 1280, eps=1e-6),
105	TemporalBlock(8, 160, 1280, eps=1e-6),
106	TemporalBlock(8, 160, 1280, eps=1e-6),
107	TemporalBlock(8, 160, 1280, eps=1e-6),
108	TemporalBlock(8, 160, 1280, eps=1e-6),
109	TemporalBlock(8, 160, 1280, eps=1e-6),
110	TemporalBlock(8, 160, 1280, eps=1e-6),
111	TemporalBlock(8, 160, 1280, eps=1e-6),
112	TemporalBlock(8, 160, 1280, eps=1e-6),
113	TemporalBlock(8, 80, 640, eps=1e-6),
114	TemporalBlock(8, 80, 640, eps=1e-6),
115	TemporalBlock(8, 80, 640, eps=1e-6),
116	TemporalBlock(8, 40, 320, eps=1e-6),
117	TemporalBlock(8, 40, 320, eps=1e-6),
118	TemporalBlock(8, 40, 320, eps=1e-6),
119	])
120	self.call_block_id = {
121	1: 0,
122	4: 1,
123	9: 2,
124	12: 3,
125	17: 4,
126	20: 5,
127	24: 6,
128	26: 7,
129	29: 8,
130	32: 9,
131	34: 10,
132	36: 11,
133	40: 12,
134	43: 13,
135	46: 14,
136	50: 15,
137	53: 16,
138	56: 17,
139	60: 18,
140	63: 19,
141	66: 20
142	}
143
144	def forward(self):
145	pass

nothing calls this directly

TemporalBlockClass · 0.85

__init__Method · 0.45

no test coverage detected