Skip to content

Update to gotch 0.9.0 #10

@jdrake

Description

@jdrake

Would be great to update transformer to work with https://github.com/sugarme/gotch/tree/v0.9.0

For example in /transformer/util/activation.go, the MustGelu method signature changed to accept a string and a bool:

func (g GeluActivation) Fwd(x *ts.Tensor) (retVal *ts.Tensor) {
	return x.MustGelu("???", false)
                           ^^^
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions