Fix: Working on improving the model code to get a better learning rate than 2.5

This commit is contained in:
Dan
2024-05-17 23:33:02 -04:00
parent 763514e815
commit 47c8cce3dd
3 changed files with 76 additions and 40 deletions

View File

@ -1,13 +1,14 @@
import os
import discord
from train_gpt_model import process_message
from gpt_model import load_model
import torch
from dotenv import load_dotenv
import os
from train_gpt_model import process_message
from gpt_model import load_model
# Load environment variables from .env file
load_dotenv()
# Discord bot token
# Get the Discord bot token from environment variables
TOKEN = os.getenv("DISCORD_TOKEN")
# Load the vocabulary