Allow .model extension for tokenizers. (pytorch#3637)

Summary: Pull Request resolved: pytorch#3637 . Reviewed By: kirklandsign Differential Revision: D57446798 fbshipit-source-id: 86c05d810d8795fe4732fc2ab649d745b6ad45be
kirklandsign · May 16, 2024 · 29f8326 · 29f8326
1 parent 121b04c
commit 29f8326
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/demo-apps/apple_ios/LLaMA/LLaMA/Application/ContentView.swift b/examples/demo-apps/apple_ios/LLaMA/LLaMA/Application/ContentView.swift
@@ -255,7 +255,7 @@ struct ContentView: View {
     case .model:
       return [UTType(filenameExtension: "pte")].compactMap { $0 }
     case .tokenizer:
-      return [UTType(filenameExtension: "bin")].compactMap { $0 }
+      return [UTType(filenameExtension: "bin"), UTType(filenameExtension: "model")].compactMap { $0 }
     }
   }