(device)
| 62 | |
| 63 | |
| 64 | def move_to(device): |
| 65 | global _llm,_tokenizer |
| 66 | if _llm is None: |
| 67 | raise ValueError("Error, LLM is not initialized.") |
| 68 | _llm = _llm.to(device) |
| 69 | if _tokenizer is None: |
| 70 | raise ValueError("Error, Tokenizer is not initialized.") |
| 71 | return f"Moved to {device}" |
| 72 | |
| 73 | # ======= Preprocessing: Merge Model |
| 74 | def merge_weight(): |
nothing calls this directly
no outgoing calls
no test coverage detected