mirror of
https://github.com/deepseek-ai/DeepSeek-V3.git
synced 2025-02-23 14:18:57 -05:00
89882a94f6
Changes: init_distributed function: Extracted the distributed setup logic into a separate function. sample function: Modified it to use torch.multinomial instead of an exponentiation-based approach for sampling. Argument Validation: Replaced the assert with a more user-friendly validation in main to ensure that at least one of the parameters (input-file or interactive) is provided. Interactive Code Refactoring: The user interaction logic was kept, but the init_distributed function is now called separately at the beginning of main. |
||
|---|---|---|
| .. | ||
| configs | ||
| convert.py | ||
| fp8_cast_bf16.py | ||
| generate.py | ||
| kernel.py | ||
| model.py | ||
| requirements.txt | ||