rm typer dep

Author: [email protected]

src/demo_hf.py

@@ -120,20 +120,12 @@ def generate(
 
 
 def main(prompt: str = "my name is", model_name: str = "blt-1b"):
-    # distributed_args = DistributedArgs()
-    # distributed_args.configure_world()
-    # if not torch.distributed.is_initialized():
-    #     setup_torch_distributed(distributed_args)
-
     # Set device and ensure CUDA is available
     if not torch.cuda.is_available():
         raise RuntimeError("CUDA is required but not available")
     device = torch.device("cuda")
     torch.cuda.empty_cache()  # Clear any existing CUDA memory
 
-    assert model_name in ["blt-1b", "blt-7b"]
-    model_name = model_name.replace("-", "_")
-
     #HF
     blt_repo = "facebook/blt-1b"
 
@@ -197,4 +189,4 @@ def main(prompt: str = "my name is", model_name: str = "blt-1b"):
 
 
 if __name__ == "__main__":
-    typer.run(main)
+    main()