Dejiao Z commited on
Commit ·
3c2fe87
1
Parent(s): 2002840
updated readme
Browse files
README.md
CHANGED
|
@@ -44,6 +44,7 @@ model = AutoModel.from_pretrained(checkpoint, trust_remote_code=True).to(device)
|
|
| 44 |
code_snippet = "def print_hello_world():\tprint('Hello World!')"
|
| 45 |
inputs = tokenizer.encode(code_snippet, return_tensors="pt").to(device)
|
| 46 |
embedding = model(inputs)[0] # Extract the embedding
|
|
|
|
| 47 |
|
| 48 |
### **Code Retrieval Performance**
|
| 49 |
|
|
|
|
| 44 |
code_snippet = "def print_hello_world():\tprint('Hello World!')"
|
| 45 |
inputs = tokenizer.encode(code_snippet, return_tensors="pt").to(device)
|
| 46 |
embedding = model(inputs)[0] # Extract the embedding
|
| 47 |
+
```
|
| 48 |
|
| 49 |
### **Code Retrieval Performance**
|
| 50 |
|