PhotoMaker-V2-CPU

Runtime error

Paper99 commited on Jul 22

Commit

fb796b9

•

1 Parent(s): e16bad8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ ASPECT_RATIO_LABELS = list(aspect_ratios)
 DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
 enable_doodle_arg = False
-photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker", filename="photomaker-v2.bin", repo_type="model")
 if torch.cuda.is_available() and torch.cuda.is_bf16_supported():
     torch_dtype = torch.bfloat16

 DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
 enable_doodle_arg = False
+photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker-V2", filename="photomaker-v2.bin", repo_type="model")
 if torch.cuda.is_available() and torch.cuda.is_bf16_supported():
     torch_dtype = torch.bfloat16