sam2ai commited on
Commit
5ce70ac
1 Parent(s): 0d010d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md CHANGED
@@ -1,3 +1,30 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+
5
+ ```
6
+ from transformers import AutoTokenizer, AutoModelForCausalLM
7
+ import transformers
8
+ import torch
9
+
10
+ model = "sam2ai/falcon-base-1b-odia-pt"
11
+
12
+ tokenizer = AutoTokenizer.from_pretrained(model)
13
+ pipeline = transformers.pipeline(
14
+ "text-generation",
15
+ model=model,
16
+ tokenizer=tokenizer,
17
+ torch_dtype=torch.bfloat16,
18
+ device_map="auto",
19
+ )
20
+ sequences = pipeline(
21
+ "ମୁଁ ହେବାକୁ ଚାହେଁ",
22
+ max_length=1024,
23
+ do_sample=True,
24
+ top_k=10,
25
+ num_return_sequences=1,
26
+ eos_token_id=tokenizer.eos_token_id,
27
+ )
28
+ for seq in sequences:
29
+ print(f"Result: {seq['generated_text']}")
30
+ ```