[Bug]: 500 is thrown upon checking status

Is there an existing issue for this?

[X] I have searched the existing issues and checked the recent builds/commits of both this extension and the webui

What happened?

{
  "error": "AttributeError",
  "detail": "",
  "body": "",
  "errors": "'DreamState' object has no attribute 'in_progress_epoch'"
}

Above is an error when I try to get a status of dreambooth while Automatic1111 is training dreambooth. I am using the latest stable version which is 1.0.14.

Steps to reproduce the problem

call /dreambooth/status

Commit and libraries

Starting at Initializing Dreambooth and ending several lines below at [+] bitsandbytes version 0.35.4 installed..

Command Line Arguments

Console logs

Traceback (most recent call last):
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\anyio\streams\memory.py", line 98, in receive
        return self.receive_nowait()
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\anyio\streams\memory.py", line 93, in receive_nowait
        raise WouldBlock
    anyio.WouldBlock

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 78, in call_next
        message = await recv_stream.receive()
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\anyio\streams\memory.py", line 118, in receive
        raise EndOfStream
    anyio.EndOfStream

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\modules\api\api.py", line 152, in exception_handling
        return await call_next(request)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 84, in call_next
        raise app_exc
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 70, in coro
        await self.app(scope, receive_or_disconnect, send_no_error)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 108, in __call__
        response = await self.dispatch_func(request, call_next)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\modules\api\api.py", line 116, in log_and_time
        res: Response = await call_next(req)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 84, in call_next
        raise app_exc
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\base.py", line 70, in coro
        await self.app(scope, receive_or_disconnect, send_no_error)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\cors.py", line 84, in __call__
        await self.app(scope, receive, send)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\gzip.py", line 24, in __call__
        await responder(scope, receive, send)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\gzip.py", line 44, in __call__
        await self.app(scope, receive, self.send_with_gzip)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\exceptions.py", line 79, in __call__
        raise exc
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\middleware\exceptions.py", line 68, in __call__
        await self.app(scope, receive, sender)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\fastapi\middleware\asyncexitstack.py", line 21, in __call__
        raise e
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\fastapi\middleware\asyncexitstack.py", line 18, in __call__
        await self.app(scope, receive, send)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\routing.py", line 718, in __call__
        await route.handle(scope, receive, send)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\routing.py", line 276, in handle
        await self.app(scope, receive, send)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\starlette\routing.py", line 66, in app
        response = await func(request)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\fastapi\routing.py", line 237, in app
        raw_response = await run_endpoint_function(
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\venv\lib\site-packages\fastapi\routing.py", line 163, in run_endpoint_function
        return await dependant.call(**values)
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\extensions\sd_dreambooth_extension\scripts\api.py", line 687, in check_status
        return JSONResponse(content={"current_state": f"{json.dumps(shared.status.dict())}"})
      File "C:\Users\Dev\Developers\StableDiffusion\stable-diffusion-webui\extensions\sd_dreambooth_extension\dreambooth\shared.py", line 202, in dict
        "in_progress_epoch": self.in_progress_epoch,
    AttributeError: 'DreamState' object has no attribute 'in_progress_epoch'

Additional information

No response

d8ahazard / sd_dreambooth_extension

[Bug]: 500 is thrown upon checking status #1346