Skip to content

Commit

Permalink
minor cleanup
Browse files Browse the repository at this point in the history
Signed-off-by: Dave Lee <[email protected]>
  • Loading branch information
dave-gray101 committed Aug 21, 2024
1 parent aa7754a commit b15b275
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
3 changes: 1 addition & 2 deletions backend/python/transformers-musicgen/backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,8 +69,7 @@ def SoundGeneration(self, request, context):
return backend_pb2.Result(success=False, message="request.model is required")
try:
self.processor = AutoProcessor.from_pretrained(model_name)
self.model = MusicgenForConditionalGeneration.from_pretrained(model_name)
print("[transformers-musicgen] SoundGeneration REQUEST: ", request, file=sys.stderr)
self.model = MusicgenForConditionalGeneration.from_pretrained(model_name)s
inputs = None
if request.text == "":
inputs = self.model.get_unconditional_inputs(num_samples=1)
Expand Down
2 changes: 0 additions & 2 deletions core/http/endpoints/elevenlabs/soundgeneration.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,6 @@ func SoundGenerationEndpoint(cl *config.BackendConfigLoader, ml *model.ModelLoad
return err
}

log.Debug().Msgf("\n\n>>> TMP after parse: %+v\n\n", input)

modelFile, err := fiberContext.ModelFromContext(c, cl, ml, input.ModelID, false)
if err != nil {
modelFile = input.ModelID
Expand Down

0 comments on commit b15b275

Please sign in to comment.