19 lines
494 B
Ruby
19 lines
494 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
class FixVllmModelName < ActiveRecord::Migration[7.1]
|
||
|
|
def up
|
||
|
|
vllm_mixtral_model_id = DB.query_single(<<~SQL).first
|
||
|
|
SELECT id FROM llm_models WHERE name = 'mistralai/Mixtral'
|
||
|
|
SQL
|
||
|
|
|
||
|
|
DB.exec(<<~SQL, target_id: vllm_mixtral_model_id) if vllm_mixtral_model_id
|
||
|
|
UPDATE llm_models
|
||
|
|
SET name = 'mistralai/Mixtral-8x7B-Instruct-v0.1'
|
||
|
|
WHERE id = :target_id
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
raise ActiveRecord::IrreversibleMigration
|
||
|
|
end
|
||
|
|
end
|