Maximofn commited on
Commit
f179ff2
·
1 Parent(s): cb57378

fix(src): :rocket: Fix model initialization

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,12 +13,12 @@ from hyvideo.constants import NEGATIVE_PROMPT
13
 
14
  def initialize_model(model_path):
15
  args = parse_args()
16
- models_root_path = Path(model_path)
17
- if not models_root_path.exists():
18
- raise ValueError(f"`models_root` not exists: {models_root_path}")
19
 
20
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained(
21
- models_root_path,
22
  args=args,
23
  device_map="auto"
24
  )
 
13
 
14
  def initialize_model(model_path):
15
  args = parse_args()
16
+ # models_root_path = Path(model_path)
17
+ # if not models_root_path.exists():
18
+ # raise ValueError(f"`models_root` not exists: {models_root_path}")
19
 
20
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained(
21
+ model_path,
22
  args=args,
23
  device_map="auto"
24
  )