Views
No views yet
1!pip install git+https://github.com/Hajime-Y/moment.git
2!pip install -U transformers
3!git clone https://github.com/Hajime-Y/Mists.git1import torch
2
3from Mists.configuration_mists import MistsConfig
4from Mists.modeling_mists import MistsForConditionalGeneration
5from Mists.processing_mists import MistsProcessor
6
7model_id = "HachiML/Mists-7B-v0.1-not-trained"
8model = MistsForConditionalGeneration.from_pretrained(
9 model_id,
10 torch_dtype=torch.bfloat16,
11 low_cpu_mem_usage=True,
12).to("cuda")
13processor = MistsProcessor.from_pretrained(model_id)1import pandas as pd
2
3hist_ndaq_512 = pd.DataFrame("nasdaq_price_history.csv")
4time_series_data = torch.tensor(hist_ndaq_512[["Open", "High", "Low", "Close", "Volume"]].values, dtype=torch.float)
5time_series_data = time_series_data.t().unsqueeze(0)
6
7prompt = "USER: <time_series>\nWhat are the features of this data?\nASSISTANT:"
8inputs = processor(prompt, time_series_data, return_tensors='pt').to(torch.float32)
9
10output = model.generate(**inputs, max_new_tokens=200, do_sample=False)
11print(processor.decode(output[0], skip_special_tokens=True))