Aarifkhan commited on
Commit
95f73fd
1 Parent(s): 2f203e6

Updated readme of vortex 3b v2

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -28,3 +28,13 @@ generated_text = pipe(text, max_length=100, do_sample=True)[0]['generated_text']
28
 
29
  print(generated_text)
30
  ```
 
 
 
 
 
 
 
 
 
 
 
28
 
29
  print(generated_text)
30
  ```
31
+
32
+ ```python
33
+ # Use a pipeline as a high-level helper
34
+ from transformers import pipeline
35
+
36
+ text = pipeline(model="OEvortex/vortex-3b-v2", torch_dtype=torch.bfloat16, device_map="auto")
37
+
38
+ res = text("Explain to me the difference between nuclear fission and fusion.")
39
+ print(res[0]["text"])
40
+ ```