Initial mirror of localai-org/motion-bricks.cpp@6fdb75e

d456972 verified about 2 months ago

1.91 kB

#pragma once

#include <motionbricks/motionbricks.h>

#include <cstdint>
#include <span>
#include <string>
#include <vector>

namespace motionbricks::detail {

class neural_runtime;

struct root_result {
    std::uint32_t tokens = 0;
    std::vector<float> duration_logits;
    std::vector<float> global_root_values;
};

mb_status run_root_planner(const neural_runtime & runtime,
                           std::span<const float> global_root_values,
                           std::span<const std::uint8_t> has_global_root_values,
                           std::span<const float> local_root_values,
                           std::span<const std::uint8_t> has_local_root_values,
                           std::span<const float> poses,
                           std::span<const std::uint8_t> has_poses,
                           std::uint32_t requested_tokens,
                           root_result & output,
                           std::string & reason);

/* Uses the masked-duration embedding (upstream token value 18) while building
   a graph for candidate_tokens. This is the first pass of auto-duration. */
mb_status run_root_planner_auto_probe(const neural_runtime & runtime,
                                      std::span<const float> global_root_values,
                                      std::span<const std::uint8_t> has_global_root_values,
                                      std::span<const float> local_root_values,
                                      std::span<const std::uint8_t> has_local_root_values,
                                      std::span<const float> poses,
                                      std::span<const std::uint8_t> has_poses,
                                      std::uint32_t candidate_tokens,
                                      root_result & output,
                                      std::string & reason);

} // namespace motionbricks::detail

	#pragma once

	#include <motionbricks/motionbricks.h>

	#include <cstdint>
	#include <span>
	#include <string>
	#include <vector>

	namespace motionbricks::detail {

	class neural_runtime;

	struct root_result {
	std::uint32_t tokens = 0;
	std::vector<float> duration_logits;
	std::vector<float> global_root_values;
	};

	mb_status run_root_planner(const neural_runtime & runtime,
	std::span<const float> global_root_values,
	std::span<const std::uint8_t> has_global_root_values,
	std::span<const float> local_root_values,
	std::span<const std::uint8_t> has_local_root_values,
	std::span<const float> poses,
	std::span<const std::uint8_t> has_poses,
	std::uint32_t requested_tokens,
	root_result & output,
	std::string & reason);

	/* Uses the masked-duration embedding (upstream token value 18) while building
	a graph for candidate_tokens. This is the first pass of auto-duration. */
	mb_status run_root_planner_auto_probe(const neural_runtime & runtime,
	std::span<const float> global_root_values,
	std::span<const std::uint8_t> has_global_root_values,
	std::span<const float> local_root_values,
	std::span<const std::uint8_t> has_local_root_values,
	std::span<const float> poses,
	std::span<const std::uint8_t> has_poses,
	std::uint32_t candidate_tokens,
	root_result & output,
	std::string & reason);

	} // namespace motionbricks::detail