Skip to content

Commit

Permalink
chore: prepare for release
Browse files Browse the repository at this point in the history
  • Loading branch information
sno2 committed Nov 21, 2021
1 parent 7df2dbc commit 6617746
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 9 deletions.
6 changes: 3 additions & 3 deletions Cargo.toml
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
[package]
name = "libffi"
name = "bertml"
version = "0.1.0"
edition = "2021"

[lib]
name = "libffi"
name = "bertml"
crate-type = ["cdylib"]

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
Expand All @@ -15,4 +15,4 @@ rust-bert = { git = "https://github.com/guillaume-be/rust-bert.git" }
serde = { version = "1", features = ["derive"] }
serde_json = "1.0"
anyhow = "1.0"
uuid = { version = "0.8.2", features = ["v4"] } # matches rust-bert's
uuid = { version = "0.8.2", features = ["v4"] } # matches rust-bert's
10 changes: 4 additions & 6 deletions model_manager.ts
Original file line number Diff line number Diff line change
Expand Up @@ -131,8 +131,8 @@ const BINARY_LOCATION = (() => {
linux: ["lib", "so"],
}[Deno.build.os];

const name = "libffi";
const path = "./target/release/";
const name = "bertml";
const path = "https://github.com/sno2/bertml/releases/download/v0.1.0-alpha/";

return `${path}${prefix}${name}.${ext}`;
})();
Expand Down Expand Up @@ -214,7 +214,7 @@ export class ModelManager {
}

async createTranslationModel<T extends TranslationModelInit>(
init: T,
init: T
): Promise<TranslationModel<T>> {
const bytes = encode(JSON.stringify(init));
const rid = await this.bindings
Expand All @@ -241,9 +241,7 @@ export class ModelManager {
return model;
}

async createZeroShotClassificationModel(): Promise<
ZeroShotClassificationModel
> {
async createZeroShotClassificationModel(): Promise<ZeroShotClassificationModel> {
const rid = await this.bindings
.create_zero_shot_model()
.then(this.assertCode);
Expand Down

0 comments on commit 6617746

Please sign in to comment.