Spaces:
Paused
Paused
koby commited on
Commit ·
ba46d02
1
Parent(s): 9392b1b
fix TypeError: drop trust_remote_code (not accepted by YuE2Pipeline)
Browse files
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
|