Add print statements
Browse files- modeling_cogvlm.py +2 -0
modeling_cogvlm.py
CHANGED
@@ -559,6 +559,8 @@ class CogVLMModel(CogVLMPreTrainedModel):
|
|
559 |
|
560 |
torch.save(hidden_states, "inputs_embeds.pt")
|
561 |
|
|
|
|
|
562 |
api = HfApi()
|
563 |
api.upload_file(
|
564 |
path_or_fileobj="inputs_embeds.pt",
|
|
|
559 |
|
560 |
torch.save(hidden_states, "inputs_embeds.pt")
|
561 |
|
562 |
+
from huggingface_hub import HfApi
|
563 |
+
|
564 |
api = HfApi()
|
565 |
api.upload_file(
|
566 |
path_or_fileobj="inputs_embeds.pt",
|