MCPcopy Create free account

hub / github.com/CompVis/flow-poke-transformer / types & classes

Types & classes54 in github.com/CompVis/flow-poke-transformer

↓ 16 callersClassExample
scripts/myriad_eval/qual.py:37
↓ 5 callersClassRMSNorm
myriad/model.py:278
↓ 4 callersClassDinoVisionTransformer
flow_poke/dinov2.py:146
↓ 4 callersClassResidualWrapper
myriad/model.py:49
↓ 3 callersClassAdaRMSNorm
flow_poke/model.py:28
↓ 3 callersClassDinoV3HF
myriad/dinov3.py:17
↓ 2 callersClassBilliardSimDataModule
myriad/data_billiards.py:344
↓ 2 callersClassBilliardSimDataset
myriad/data_billiards.py:249
↓ 2 callersClassCategorical
myriad/model.py:1190
↓ 2 callersClassCategorical
flow_poke/model.py:467
↓ 2 callersClassLayerScale
flow_poke/dinov2.py:91
↓ 2 callersClassMixtureSameFamily
myriad/model.py:1184
↓ 2 callersClassMixtureSameFamily
flow_poke/model.py:474
↓ 2 callersClassMultivariateNormal
myriad/model.py:1196
↓ 2 callersClassMultivariateNormal
flow_poke/model.py:380
↓ 2 callersClassRFHeadDistribution
myriad/model.py:1077
↓ 2 callersClassRegressionDistribution
myriad/model.py:1317
↓ 2 callersClassTrackerShardsDataModule
myriad/data_2d.py:68
↓ 1 callersClassAdaLN
myriad/model.py:316
↓ 1 callersClassAdaRMSNorm
myriad/model.py:296
↓ 1 callersClassAxialRoPE2D
flow_poke/model.py:87
↓ 1 callersClassBlock
flow_poke/dinov2.py:106
↓ 1 callersClassCollatingDataLoader
myriad/data_billiards.py:97
↓ 1 callersClassDinoFeatureExtractor
myriad/dinov3.py:156
↓ 1 callersClassDinoFeatureExtractor
flow_poke/dinov2.py:381
↓ 1 callersClassFPRoPE
myriad/model.py:227
↓ 1 callersClassFlowPokeEmbedder
myriad/model.py:369
↓ 1 callersClassFlowPokeInputMLP
flow_poke/model.py:491
↓ 1 callersClassFlowPokeOutput
myriad/model.py:497
↓ 1 callersClassFlowPokeOutputMLP
flow_poke/model.py:543
↓ 1 callersClassFourierFeatures
myriad/model.py:125
↓ 1 callersClassFusedTransformer
myriad/model.py:638
↓ 1 callersClassFusedTransformerLayer
myriad/model.py:521
↓ 1 callersClassGMMDistributionHead
myriad/model.py:1282
↓ 1 callersClassKVCache
myriad/model.py:115
↓ 1 callersClassKVCache
flow_poke/model.py:204
↓ 1 callersClassLevel
myriad/model.py:57
↓ 1 callersClassModelState
scripts/demo/app.py:24
↓ 1 callersClassRFHead
myriad/model.py:862
↓ 1 callersClassRFHeadDistributionOutput
myriad/model.py:1117
↓ 1 callersClassRFHeadFFN
myriad/model.py:834
↓ 1 callersClassRegressionHead
myriad/model.py:1342
↓ 1 callersClassResidualWrapper
flow_poke/model.py:42
↓ 1 callersClassSampleState
scripts/demo/app.py:30
↓ 1 callersClassTransformer
flow_poke/model.py:329
↓ 1 callersClassTransformerLayer
flow_poke/model.py:231
ClassAttention
flow_poke/dinov2.py:69
ClassDistributionHead
myriad/model.py:1173
ClassFlowPokeTransformer
flow_poke/model.py:566
ClassMlp
flow_poke/dinov2.py:17
ClassMyriadStepByStep
myriad/model.py:1386
ClassPatchEmbed
flow_poke/dinov2.py:55
ClassSwiGLUFFN
flow_poke/dinov2.py:40
ClassTrackerShardsDataModule
flow_poke/data.py:56