mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-08 21:38:04 -05:00
68e9281778
* Shark Studio SDXL support, HIP driver support, simpler device info, small fixes * Fixups to llm API/UI and ignore user config files. * Small fixes for unifying pipelines. * Update requirements.txt for iree-turbine (#2130) * Fix Llama2 on CPU (#2133) * Filesystem cleanup and custom model fixes (#2127) * Fix some formatting issues * Remove IREE pin (fixes exe issue) (#2126) * Update find links for IREE packages (#2136) * Shark Studio SDXL support, HIP driver support, simpler device info, small fixes * Abstract out SD pipelines from Studio Webui (WIP) * Switch from pin to minimum torch version and fix index url * Fix device parsing. * Fix linux setup * Fix custom weights. --------- Co-authored-by: saienduri <77521230+saienduri@users.noreply.github.com> Co-authored-by: gpetters-amd <159576198+gpetters-amd@users.noreply.github.com> Co-authored-by: gpetters94 <gpetters@protonmail.com>
96 lines
2.5 KiB
PowerShell
96 lines
2.5 KiB
PowerShell
<#
|
|
.SYNOPSIS
|
|
A script to update and install the SHARK runtime and its dependencies.
|
|
|
|
.DESCRIPTION
|
|
This script updates and installs the SHARK runtime and its dependencies.
|
|
It checks the Python version installed and installs any required build
|
|
dependencies into a Python virtual environment.
|
|
If that environment does not exist, it creates it.
|
|
|
|
.PARAMETER update-src
|
|
git pulls latest version
|
|
|
|
.PARAMETER force
|
|
removes and recreates venv to force update of all dependencies
|
|
|
|
.EXAMPLE
|
|
.\setup_venv.ps1 --force
|
|
|
|
.EXAMPLE
|
|
.\setup_venv.ps1 --update-src
|
|
|
|
.INPUTS
|
|
None
|
|
|
|
.OUTPUTS
|
|
None
|
|
|
|
#>
|
|
|
|
param([string]$arguments)
|
|
|
|
if ($arguments -eq "--update-src"){
|
|
git pull
|
|
}
|
|
|
|
if ($arguments -eq "--force"){
|
|
if (Test-Path env:VIRTUAL_ENV) {
|
|
Write-Host "deactivating..."
|
|
Deactivate
|
|
}
|
|
|
|
if (Test-Path .\shark.venv\) {
|
|
Write-Host "removing and recreating venv..."
|
|
Remove-Item .\shark.venv -Force -Recurse
|
|
if (Test-Path .\shark.venv\) {
|
|
Write-Host 'could not remove .\shark-venv - please try running ".\setup_venv.ps1 --force" again!'
|
|
exit 1
|
|
}
|
|
}
|
|
}
|
|
|
|
# redirect stderr into stdout
|
|
$p = &{python -V} 2>&1
|
|
# check if an ErrorRecord was returned
|
|
$version = if($p -is [System.Management.Automation.ErrorRecord])
|
|
{
|
|
# grab the version string from the error message
|
|
$p.Exception.Message
|
|
}
|
|
else
|
|
{
|
|
# otherwise return complete Python list
|
|
$ErrorActionPreference = 'SilentlyContinue'
|
|
$PyVer = py --list
|
|
}
|
|
|
|
# deactivate any activated venvs
|
|
if ($PyVer -like "*venv*")
|
|
{
|
|
deactivate # make sure we don't update the wrong venv
|
|
$PyVer = py --list # update list
|
|
}
|
|
|
|
Write-Host "Python versions found are"
|
|
Write-Host ($PyVer | Out-String) # formatted output with line breaks
|
|
if (!($PyVer.length -ne 0)) {$p} # return Python --version String if py.exe is unavailable
|
|
if (!($PyVer -like "*3.11*") -and !($p -like "*3.11*")) # if 3.11 is not in any list
|
|
{
|
|
Write-Host "Please install Python 3.11 and try again"
|
|
exit 34
|
|
}
|
|
|
|
Write-Host "Installing Build Dependencies"
|
|
# make sure we really use 3.11 from list, even if it's not the default.
|
|
if ($NULL -ne $PyVer) {py -3.11 -m venv .\shark.venv\}
|
|
else {python -m venv .\shark.venv\}
|
|
.\shark.venv\Scripts\activate
|
|
python -m pip install --upgrade pip
|
|
pip install wheel
|
|
pip install --pre -r requirements.txt
|
|
pip install -e .
|
|
|
|
>>>>>>> 0c904eb7 (Shark Studio SDXL support, HIP driver support, simpler device info, small fixes)
|
|
Write-Host "Source your venv with ./shark.venv/Scripts/activate"
|