loadFromDirectory method

Future<void> loadFromDirectory(String directory)

Load weights from a directory.

Searches in order:

1. model.safetensors
2. weights.json

Implementation

Future<void> loadFromDirectory(String directory) async {
  final safetensorsFile = File('$directory/model.safetensors');
  if (await safetensorsFile.exists()) {
    final weights = await _loadSafetensors(safetensorsFile);
    loadFromMap(weights);
    return;
  }

  final jsonFile = File('$directory/weights.json');
  if (await jsonFile.exists()) {
    final weights = await _loadJsonWeights(jsonFile);
    loadFromMap(weights);
    return;
  }

  throw FileSystemException(
    'No weight file found in directory. '
    'Expected model.safetensors or weights.json',
    directory,
  );
}