diff --git a/lib/core/llm/mediapipe_llm_service.dart b/lib/core/llm/mediapipe_llm_service.dart index 347ba7c..29808cb 100644 --- a/lib/core/llm/mediapipe_llm_service.dart +++ b/lib/core/llm/mediapipe_llm_service.dart @@ -7,7 +7,7 @@ class MediapipeLlmService implements LlmService { final _chat = FlutterMediapipeChat(); bool _initialised = false; - static const _modelFileName = 'gemma-2b-it-cpu-int4.bin'; + static const _modelFileName = 'gemma-2b-it-gpu-int4.bin'; @override Future initialise() async { diff --git a/lib/core/llm/model_download_service.dart b/lib/core/llm/model_download_service.dart index fea01b5..1efbeeb 100644 --- a/lib/core/llm/model_download_service.dart +++ b/lib/core/llm/model_download_service.dart @@ -3,7 +3,7 @@ import 'package:path_provider/path_provider.dart'; import 'package:path/path.dart' as p; class ModelDownloadService { - static const modelFileName = 'gemma-2b-it-cpu-int4.bin'; + static const modelFileName = 'gemma-2b-it-gpu-int4.bin'; static const modelSizeBytes = 1183285248; // ~1.1 GB // Gemma 2B IT CPU int4 — MediaPipe LLM Inference model diff --git a/lib/features/model_download/model_download_screen.dart b/lib/features/model_download/model_download_screen.dart index 08cd5fc..b80e541 100644 --- a/lib/features/model_download/model_download_screen.dart +++ b/lib/features/model_download/model_download_screen.dart @@ -91,13 +91,13 @@ class _InstructionCard extends StatelessWidget { const SizedBox(height: 10), _Step( n: '1', - text: 'Download gemma-2b-it-cpu-int4.bin from Kaggle ' + text: 'Download gemma-2b-it-gpu-int4.bin from Kaggle ' '(requires a free Google account and licence acceptance).', ), _Step( n: '2', text: 'Transfer the file to the device using adb:\n' - 'adb push gemma-2b-it-cpu-int4.bin ' + 'adb push gemma-2b-it-gpu-int4.bin ' r'$(adb shell run-as net.stefwill.dailyou ' r'printenv DATA_DIR)/files/', ),