getModelFilePaths method

  1. @override
Future<Map<String, String>?> getModelFilePaths(
  1. ModelSpec spec
)
override

Gets the file paths for an installed model

Implementation

@override
Future<Map<String, String>?> getModelFilePaths(ModelSpec spec) async {
  await _ensureInitialized();

  if (!await isModelInstalled(spec)) {
    return null;
  }

  final paths = <String, String>{};
  final modelPath = _modelPaths[spec.name];
  final loraPath = _loraPaths[spec.name];

  if (modelPath != null) {
    paths['model'] = modelPath;
  }
  if (loraPath != null) {
    paths['lora'] = loraPath;
  }

  return paths.isNotEmpty ? paths : null;
}