ElatedBat21

1 Question, 1 Answer

Active since 21 November 2023

Last activity 5 months ago

Reputation

Badges 1

Eureka!

Questions 1
Answers 1

0 Votes

2 Answers

312 Views

0 Votes 2 Answers 312 Views

Hello Everyone, Is The Logging Of "Automodelforcausallm" Models Supported? Even With Manually Logging Of Llama-2-7B-Hf It Does Not Appear In The Model List (Even When Uploading As Artifact)

Hello everyone, is the logging of "AutoModelForCausalLM" models supported? Even with manually logging of Llama-2-7b-hf it does not appear in the model list (...

clearml

6 months ago

0 Hello Everyone, Is The Logging Of "Automodelforcausallm" Models Supported? Even With Manually Logging Of Llama-2-7B-Hf It Does Not Appear In The Model List (Even When Uploading As Artifact)

Hi @<1523701070390366208:profile|CostlyOstrich36>

Thanks for the quick reply.

Here is a snipped of the code I use to register the model.
I load in the model to set the parameters and try to register it.
The result is a model pickle file in the artifacts tab but no registration of the model.

task_name = "Llama2_model"
task = Task.init(project_name='project_name', task_name=task_name, output_uri="s3://...")


model = AutoModelForCausalLM.from_pretrained(Llama2_source,
                ...

5 months ago