Spaces:

Leon4gr45
/

phi_inference

Runtime error

Leon4gr45 commited on Jan 11

Commit

6947e38

verified ·

1 Parent(s): dcfb6cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model = AutoModel.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
-    revision="65b9340"
 ).to(device).eval()
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)

     trust_remote_code=True,
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
+    # revision="65b9340"
 ).to(device).eval()
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)