zstanjj
/

HTML-Pruner-Phi-3.8B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

zstanjj commited on Dec 24, 2024

Commit

9c01983

·

verified ·

1 Parent(s): e9cf3e9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -201,7 +201,7 @@ if torch.cuda.is_available():
 else:
     device="cpu"
 gen_embed_pruner = GenHTMLPruner(gen_model=ckpt_path, max_node_words=MAX_NODE_WORDS_GEN, device=device)
-block_rankings = gen_embed_pruner.calculate_block_rankings(question, pruned_html)
 print(block_rankings)
 # [1, 0]

 else:
     device="cpu"
 gen_embed_pruner = GenHTMLPruner(gen_model=ckpt_path, max_node_words=MAX_NODE_WORDS_GEN, device=device)
+block_rankings=gen_embed_pruner.calculate_block_rankings(question, pruned_html, block_tree)
 print(block_rankings)
 # [1, 0]