dfalbel commited on
Commit
35d3c36
1 Parent(s): 33145ed

Early return if generate is called and the session is not yet started.

Browse files
Files changed (1) hide show
  1. model-session.R +4 -0
model-session.R CHANGED
@@ -29,6 +29,10 @@ model_session <- R6::R6Class(
29
  })
30
  },
31
  generate = function(prompt) {
 
 
 
 
32
  args <- list(
33
  prompt = prompt,
34
  temperature = self$temperature,
 
29
  })
30
  },
31
  generate = function(prompt) {
32
+ if (is.null(self$sess)) {
33
+ cat("Model is not loaded, error.", "\n")
34
+ return(self$sess$call(function() stop("Model is not loaded")))
35
+ }
36
  args <- list(
37
  prompt = prompt,
38
  temperature = self$temperature,