main

igecloudsdev.code-workspace

@@ -4,5 +4,7 @@
  "path": "."
  }
  ],
- "settings": {}
+ "settings": {
+ "npm.enableRunFromFolder": true
+ }
 }

transformers

@@ -0,0 +1,25 @@
+from transformers import AutoTokenizer, AutoModel
+tokenizer = AutoTokenizer.from_pretrained("THUDM/codegeex2-6b", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/codegeex2-6b", trust_remote_code=True, device='cuda')
+model = model.eval()
+
+# remember adding a language tag for better performance
+prompt = "# language: Python\n# write a bubble sort function\n"
+inputs = tokenizer.encode(prompt, return_tensors="pt").to(model.device)
+outputs = model.generate(inputs, max_length=256, top_k=1)
+response = tokenizer.decode(outputs[0])
+
+>>> print(response)
+# language: Python
+# write a bubble sort function
+
+
+def bubble_sort(list):
+ for i in range(len(list) - 1):
+ for j in range(len(list) - 1):
+ if list[j] > list[j + 1]:
+ list[j], list[j + 1] = list[j + 1], list[j]
+ return list
+
+
+print(bubble_sort([5, 2, 1, 8, 4]))