Update README.md

README.md

@@ -1,3 +1,8 @@
+Galactica-6.7b finetuned on webgpt and prompt_dialogue.
+
+Demo use:
+
+```
 import torch
 from torch import nn
 from torch.nn import functional as F
@@ -33,4 +38,6 @@ with torch.cuda.amp.autocast():
         eos_token_id=tokenizer.additional_special_tokens_ids[tokenizer.additional_special_tokens.index('<question>')]
     )
 
-message = tokenizer.decode(out[0, :-1]).replace('<question>', "User:\n").replace('<answer>', 'Assistant:\n')
+message = tokenizer.decode(out[0, :-1]).replace('<question>', "User:\n").replace('<answer>', 'Assistant:\n')
+
+```