Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Lucain <[email protected]>
  • Loading branch information
FrozenBurning and Wauplin committed Sep 19, 2024
1 parent bd3a038 commit 25c98d2
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 8 deletions.
11 changes: 4 additions & 7 deletions packages/tasks/src/model-libraries-snippets.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1008,13 +1008,10 @@ whisperkit-cli transcribe --audio-path /path/to/audio.mp3
whisperkit-cli transcribe --model "large-v3" --model-prefix "distil" --audio-path /path/to/audio.mp3 --verbose`,
];

export const threedtopia_xl = (): string[] => [
`# Install from https://github.com/3DTopia/3DTopia-XL
from threedtopia_xl.models import threedtopia_xl
export const threedtopia_xl = (model: ModelData): string[] => [
`from threedtopia_xl.models import threedtopia_xl

model = threedtopia_xl.from_pretrained("${model.id}")
img = 'PATH_TO_Image.png'
model.generate(cond=img)
# refer to https://github.com/3DTopia/3DTopia-XL?tab=readme-ov-file#inference for inference
# and https://github.com/3DTopia/3DTopia-XL/blob/main/app.py regarding usage`,
model.generate(cond="path/to/image.png")
];
//#endregion
2 changes: 1 addition & 1 deletion packages/tasks/src/model-libraries.ts
Original file line number Diff line number Diff line change
Expand Up @@ -705,7 +705,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
repoName: "3DTopia-XL",
repoUrl: "https://github.com/3DTopia/3DTopia-XL",
filter: false,
countDownloads: `path_extension:"pt"`,
countDownloads: `path:"model_vae_fp16.pt"`,
snippets: snippets.threedtopia_xl,
},
} satisfies Record<string, LibraryUiElement>;
Expand Down

0 comments on commit 25c98d2

Please sign in to comment.