(SD) Fix file management in SD pipeline for parameter files.

nod-ai / SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.

Apache License 2.0

95 stars 48 forks source link

(SD) Fix file management in SD pipeline for parameter files. #857

Open monorimet opened 1 month ago

monorimet commented 1 month ago

File management needs to happen a bit more flexibly. Certain execution modes end up looking for weights in the wrong place, and we need to cleanup some hardcoding of parameter files for sharktank/shark HF sourced models like punet.