From e2cc04fce3b1a9b99e38db024edbc61adb6fa9a2 Mon Sep 17 00:00:00 2001
From: Pedro Cuenca <pedro@huggingface.co>
Date: Tue, 7 Feb 2023 21:31:27 +0100
Subject: [PATCH] Implement ModelInfo.Equatable.

---
 Diffusion/ModelInfo.swift | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Diffusion/ModelInfo.swift b/Diffusion/ModelInfo.swift
index 3ffeb26..80ed377 100644
--- a/Diffusion/ModelInfo.swift
+++ b/Diffusion/ModelInfo.swift
@@ -100,3 +100,7 @@ extension ModelInfo {
         ModelInfo.MODELS.first(where: {$0.modelId == modelId})
     }
 }
+
+extension ModelInfo : Equatable {
+    static func ==(lhs: ModelInfo, rhs: ModelInfo) -> Bool { lhs.modelId == rhs.modelId }
+}