Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
c73aad4
1
Parent(s):
1eecd9f
Update app.py
Browse filesremove torch compile
app.py
CHANGED
@@ -42,7 +42,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
42 |
config = AutoConfig.from_pretrained("nvidia/quality-classifier-deberta")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("nvidia/quality-classifier-deberta")
|
44 |
model = QualityModel.from_pretrained("nvidia/quality-classifier-deberta").to(device)
|
45 |
-
model = torch.compile(model)
|
46 |
model.eval()
|
47 |
|
48 |
|
|
|
42 |
config = AutoConfig.from_pretrained("nvidia/quality-classifier-deberta")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("nvidia/quality-classifier-deberta")
|
44 |
model = QualityModel.from_pretrained("nvidia/quality-classifier-deberta").to(device)
|
45 |
+
# model = torch.compile(model)
|
46 |
model.eval()
|
47 |
|
48 |
|