Text Generation
Transformers
Safetensors
falcon
text-generation-inference
quantumaikr commited on
Commit
d6b2220
·
1 Parent(s): 6fece32

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -0
README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - tiiuae/falcon-refinedweb
4
+ - nRuaif/wizard_alpaca_dolly_orca
5
+ language:
6
+ - en
7
+ - de
8
+ - es
9
+ - fr
10
+ inference: false
11
+ license: unknown
12
+ ---
13
+
14
+
15
+
16
+
17
+
18
+ # 🚀 quantumaikr/falcon-180B-chat-instruct
19
+
20
+ **quantumaikr/falcon-180B-chat-instruct is a 180B parameters causal decoder-only model built by [quantumaikr](https://www.quantumai.kr) based on [Falcon-180B-chat](https://huggingface.co/tiiuae/falcon-180B-chat)**
21
+
22
+
23
+
24
+ ## How to Get Started with the Model
25
+
26
+ To run inference with the model in full `bfloat16` precision you need approximately 8xA100 80GB or equivalent.
27
+
28
+
29
+ ```python
30
+ from transformers import AutoTokenizer, AutoModelForCausalLM
31
+ import transformers
32
+ import torch
33
+
34
+ model = "quantumaikr/falcon-180B-chat-instruct"
35
+
36
+ tokenizer = AutoTokenizer.from_pretrained(model)
37
+ pipeline = transformers.pipeline(
38
+ "text-generation",
39
+ model=model,
40
+ tokenizer=tokenizer,
41
+ torch_dtype=torch.bfloat16,
42
+ trust_remote_code=True,
43
+ device_map="auto",
44
+ )
45
+ sequences = pipeline(
46
+ "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
47
+ max_length=200,
48
+ do_sample=True,
49
+ top_k=10,
50
+ num_return_sequences=1,
51
+ eos_token_id=tokenizer.eos_token_id,
52
+ )
53
+ for seq in sequences:
54
+ print(f"Result: {seq['generated_text']}")
55
+
56
+ ```
57
+
58
+
59
+ ## Contact
60