diffusion-rs-sys 0.1.14

Rust bindings for stable-diffusion.cpp (FFI bindings)
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#ifndef __NAME_CONVERSTION_H__
#define __NAME_CONVERSTION_H__

#include <string>

#include "model.h"

bool is_cond_stage_model_name(const std::string& name);
bool is_diffusion_model_name(const std::string& name);
bool is_first_stage_model_name(const std::string& name);

std::string convert_tensor_name(std::string name, SDVersion version);

#endif  // __NAME_CONVERSTION_H__