Skip to content

Commit e726f24

Browse files
committed
No need to set @model_id
1 parent 9b1dd71 commit e726f24

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

lib/ruby_llm/providers/replicate/images.rb

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def images_url
1616
end
1717

1818
def render_image_payload(prompt, model:, **params)
19-
self.model_id = model
19+
self.model = model
2020

2121
{}.tap do |payload|
2222
payload[:webhook] = @config.replicate_webhook_url
@@ -43,9 +43,8 @@ def fetch_image_blob(url)
4343

4444
private
4545

46-
def model_id=(id)
47-
@model_id = id
48-
@model = RubyLLM::Models.find(@model_id, 'replicate')
46+
def model=(id)
47+
@model = RubyLLM::Models.find(id, 'replicate')
4948
end
5049

5150
def official_model?

0 commit comments

Comments
 (0)