koby commited on
Commit
ba46d02
·
1 Parent(s): 9392b1b

fix TypeError: drop trust_remote_code (not accepted by YuE2Pipeline)

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -58,7 +58,6 @@ def get_pipeline():
58
  pipe = YuE2Pipeline.from_pretrained(
59
  REPO,
60
  device="cuda" if torch.cuda.is_available() else "cpu",
61
- trust_remote_code=True
62
  )
63
  print("Pipeline loaded.")
64
  return pipe
 
58
  pipe = YuE2Pipeline.from_pretrained(
59
  REPO,
60
  device="cuda" if torch.cuda.is_available() else "cpu",
 
61
  )
62
  print("Pipeline loaded.")
63
  return pipe