yasu-oh commited on
Commit
bbf5c16
·
1 Parent(s): 99a3dfb

splitlines

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,7 +48,8 @@ def is_architecture_supported(architecture: str) -> Optional[bool]:
48
  response = requests.get(LLAMA_CPP_SOURCE)
49
  response.raise_for_status()
50
  content = response.text
51
- return architecture in content
 
52
  except:
53
  return None
54
 
 
48
  response = requests.get(LLAMA_CPP_SOURCE)
49
  response.raise_for_status()
50
  content = response.text
51
+ models = [line for line in content.splitlines() if "@Model.register" in line]
52
+ return any(architecture in entry for entry in models)
53
  except:
54
  return None
55