github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Unpin timm.

Browse Source
This commit is contained in:
Ean Garvey
2023-10-19 12:01:26 -05:00
parent 791553762b
commit 7a7252a8fd
2 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
apps/stable_diffusion/src/utils/stencils/zoe/__init__.py
View File

@@ -24,12 +24,6 @@ class ZoeDetector:
for chunk in r.iter_content(chunk_size=8192):
f.write(chunk)
midas = torch.hub.load(
"gpetters94/MiDaS:master",
"DPT_BEiT_L_384",
pretrained=False,
force_reload=False,
)
model = torch.hub.load(
"monorimet/ZoeDepth:torch_update",
"ZoeD_N",

2
requirements.txt
View File

@@ -39,7 +39,7 @@ sentencepiece
py-cpuinfo
tiktoken # for codegen
joblib # for langchain
timm==0.9.5 # for MiniGPT4
timm # for MiniGPT4
langchain
einops # for zoedepth