luxananda commited on
Commit
ebf78c0
1 Parent(s): 3fcb7aa

Create test123

Browse files
Files changed (1) hide show
  1. test123 +10 -0
test123 ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from transformers import pipeline
3
+
4
+ generate_text = pipeline(model="databricks/dolly-v2-12b",
5
+ torch_dtype=torch.bfloat16,
6
+ trust_remote_code=True,
7
+ device_map="auto")
8
+
9
+ generate_text("Explain to me the difference between nuclear \
10
+ fission and fusion.")