We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 45a2ad5 commit 1390951Copy full SHA for 1390951
ai_diffusion/ui/server.py
@@ -190,10 +190,11 @@ def _backend_supports(self, item: PackageItem):
190
return True
191
192
def _workload_matches(self, item: PackageItem):
193
+ archs_with_workload = (Arch.sd15, Arch.sdxl, Arch.flux, Arch.flux_k)
194
return (
195
not isinstance(item.package, ModelResource)
196
or item.package.arch in self.workloads
- or item.package.arch not in Arch.list()
197
+ or item.package.arch not in archs_with_workload
198
)
199
200
def _update_workload(self, item: PackageItem):
0 commit comments