Skip to content

Commit 1390951

Browse files
committed
Managed install: fix illustrious models not being selectable
1 parent 45a2ad5 commit 1390951

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ai_diffusion/ui/server.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,10 +190,11 @@ def _backend_supports(self, item: PackageItem):
190190
return True
191191

192192
def _workload_matches(self, item: PackageItem):
193+
archs_with_workload = (Arch.sd15, Arch.sdxl, Arch.flux, Arch.flux_k)
193194
return (
194195
not isinstance(item.package, ModelResource)
195196
or item.package.arch in self.workloads
196-
or item.package.arch not in Arch.list()
197+
or item.package.arch not in archs_with_workload
197198
)
198199

199200
def _update_workload(self, item: PackageItem):

0 commit comments

Comments
 (0)