Skip to content

LLAMA 8B and build-spec #16

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 13, 2025
Merged

LLAMA 8B and build-spec #16

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
92 changes: 47 additions & 45 deletions pallets/network/src/lib.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3611,55 +3611,57 @@ pub mod pallet {
return
}

// let subnet_id = 1;
let subnet_id = 1;

// let base_node_memory: u128 = BaseSubnetNodeMemoryMB::<T>::get();
let base_node_memory: u128 = BaseSubnetNodeMemoryMB::<T>::get();

// // --- Get min nodes based on default memory settings
// let real_min_subnet_nodes: u128 = self.memory_mb.clone() / base_node_memory;
// let mut min_subnet_nodes: u32 = MinSubnetNodes::<T>::get();
// if real_min_subnet_nodes as u32 > min_subnet_nodes {
// min_subnet_nodes = real_min_subnet_nodes as u32;
// }
// --- Get min nodes based on default memory settings
let real_min_subnet_nodes: u128 = self.memory_mb.clone() / base_node_memory;
let mut min_subnet_nodes: u32 = MinSubnetNodes::<T>::get();
if real_min_subnet_nodes as u32 > min_subnet_nodes {
min_subnet_nodes = real_min_subnet_nodes as u32;
}

// let target_subnet_nodes: u32 = (min_subnet_nodes as u128).saturating_mul(TargetSubnetNodesMultiplier::<T>::get()).saturating_div(1000000000) as u32 + min_subnet_nodes;

// let subnet_data = SubnetData {
// id: subnet_id,
// path: self.subnet_path.clone(),
// min_nodes: min_subnet_nodes,
// target_nodes: target_subnet_nodes,
// memory_mb: self.memory_mb.clone(),
// registration_blocks: MinSubnetRegistrationBlocks::<T>::get(),
// initialized: 1,
// activated: 0,
// entry_interval: 0,
// };
let target_subnet_nodes: u32 = (min_subnet_nodes as u128).saturating_mul(TargetSubnetNodesMultiplier::<T>::get()).saturating_div(1000000000) as u32 + min_subnet_nodes;

let subnet_data = SubnetData {
id: subnet_id,
path: self.subnet_path.clone(),
min_nodes: min_subnet_nodes,
target_nodes: target_subnet_nodes,
memory_mb: self.memory_mb.clone(),
registration_blocks: MinSubnetRegistrationBlocks::<T>::get(),
initialized: 1,
activated: 0,
entry_interval: 0,
};

// Increase total subnet memory
TotalSubnetMemoryMB::<T>::mutate(|n: &mut u128| *n += subnet_data.memory_mb);
// Store unique path
SubnetPaths::<T>::insert(self.subnet_path.clone(), subnet_id);
// Store subnet data
SubnetsData::<T>::insert(subnet_id, subnet_data.clone());
// Increase total subnets count
TotalSubnets::<T>::mutate(|n: &mut u32| *n += 1);

// // Increase total subnet memory
// TotalSubnetMemoryMB::<T>::mutate(|n: &mut u128| *n += subnet_data.memory_mb);
// // Store unique path
// SubnetPaths::<T>::insert(self.subnet_path.clone(), subnet_id);
// // Store subnet data
// SubnetsData::<T>::insert(subnet_id, subnet_data.clone());
// // Increase total subnets count
// TotalSubnets::<T>::mutate(|n: &mut u32| *n += 1);

// // Increase delegate stake to allow activation of subnet model
// let min_stake_balance = MinStakeBalance::<T>::get();
// // --- Get minimum subnet stake balance
// let min_subnet_stake_balance = min_stake_balance * min_subnet_nodes as u128;
// // --- Get required delegate stake balance for a subnet to have to stay live
// let mut min_subnet_delegate_stake_balance = (min_subnet_stake_balance as u128).saturating_mul(MinSubnetDelegateStakePercentage::<T>::get()).saturating_div(1000000000);

// // --- Get absolute minimum required subnet delegate stake balance
// let min_subnet_delegate_stake = MinSubnetDelegateStake::<T>::get();
// // --- Return here if the absolute minimum required subnet delegate stake balance is greater
// // than the calculated minimum requirement
// if min_subnet_delegate_stake > min_subnet_delegate_stake_balance {
// min_subnet_delegate_stake_balance = min_subnet_delegate_stake
// }
// TotalSubnetDelegateStakeBalance::<T>::insert(subnet_id, min_subnet_delegate_stake_balance);
LastSubnetRegistrationEpoch::<T>::set(1);

// Increase delegate stake to allow activation of subnet model
let min_stake_balance = MinStakeBalance::<T>::get();
// --- Get minimum subnet stake balance
let min_subnet_stake_balance = min_stake_balance * min_subnet_nodes as u128;
// --- Get required delegate stake balance for a subnet to have to stay live
let mut min_subnet_delegate_stake_balance = (min_subnet_stake_balance as u128).saturating_mul(MinSubnetDelegateStakePercentage::<T>::get()).saturating_div(1000000000);

// --- Get absolute minimum required subnet delegate stake balance
let min_subnet_delegate_stake = MinSubnetDelegateStake::<T>::get();
// --- Return here if the absolute minimum required subnet delegate stake balance is greater
// than the calculated minimum requirement
if min_subnet_delegate_stake > min_subnet_delegate_stake_balance {
min_subnet_delegate_stake_balance = min_subnet_delegate_stake
}
TotalSubnetDelegateStakeBalance::<T>::insert(subnet_id, min_subnet_delegate_stake_balance);

// // --- Initialize subnet nodes
// // Only initialize to test using subnet nodes
Expand Down
2 changes: 1 addition & 1 deletion tensorSpec.json
View file
Open in desktop

Large diffs are not rendered by default.

8 changes: 7 additions & 1 deletion tensorSpecRaw.json
View file
Open in desktop

Large diffs are not rendered by default.

Loading