Update WarBot.py

WarBot.py

@@ -65,6 +65,7 @@ def prepare_punct():
 
 def initialize():
     """ Loading the model """
+    torch.backends.quantized.engine = 'qnnpack' # Just for the specific machine architecture
     fit_checkpoint = "WarBot"
     tokenizer = AutoTokenizer.from_pretrained(fit_checkpoint)
     model = AutoModelForCausalLM.from_pretrained(fit_checkpoint)