azale-ai
/

Starstreak-7b-alpha

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

hafidhsoekma commited on Oct 31, 2023

Commit

c01aa61

·

1 Parent(s): 96e73be

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ pip3 install -U bitsandbytes==0.39.0 einops==0.6.1 sentencepiece
 ```python
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 model = AutoModelForCausalLM.from_pretrained(
     "azale-ai/Starstreak-7B",
@@ -76,7 +76,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```python
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 model = AutoModelForCausalLM.from_pretrained(
     "azale-ai/Starstreak-7B",

 ```python
 import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 model = AutoModelForCausalLM.from_pretrained(
     "azale-ai/Starstreak-7B",
 ```python
 import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 model = AutoModelForCausalLM.from_pretrained(
     "azale-ai/Starstreak-7B",