Update pretrain_inference.py

pretrain_inference.py

@@ -11,8 +11,8 @@ NUM_LAYERS = 4
 DIM_FEEDFORWARD = 512
 DROPOUT = 0.1
 MAX_LEN = 4096
-MODEL_PATH = "d:/图像/novel_model/best_model.pt"
-TOKENIZER_PATH = "d:/图像/novel_tokenizer.json"
+MODEL_PATH = "./novel_model/best_model.pt"
+TOKENIZER_PATH = "./novel_tokenizer.json"
 
 def generate_text(model, tokenizer, prompt, max_length=100, temperature=0.8, top_k=50, top_p=0.9, device="cuda"):
     """生成文本"""