Class: Informers::PretrainedConfig

Inherits:
Object
  • Object
show all
Defined in:
lib/informers/configs.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(config_json) ⇒ PretrainedConfig

Returns a new instance of PretrainedConfig.



3
4
5
 
# File 'lib/informers/configs.rb', line 3

def initialize(config_json)
  @config_json = config_json.to_h
end

Class Method Details

.from_pretrained(pretrained_model_name_or_path, progress_callback: nil, config: nil, cache_dir: nil, local_files_only: false, revision: "main", **kwargs) ⇒ Object 



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/informers/configs.rb', line 19

def self.from_pretrained(
  pretrained_model_name_or_path,
  progress_callback: nil,
  config: nil,
  cache_dir: nil,
  local_files_only: false,
  revision: "main",
  **kwargs
)
  data = config || load_config(
    pretrained_model_name_or_path,
    progress_callback:,
    config:,
    cache_dir:,
    local_files_only:,
    revision:
  )
  new(data)
end

.load_config(pretrained_model_name_or_path, **options) ⇒ Object 



39
40
41
42
 
# File 'lib/informers/configs.rb', line 39

def self.load_config(pretrained_model_name_or_path, **options)
  info = Utils::Hub.get_model_json(pretrained_model_name_or_path, "config.json", true, **options)
  info
end

Instance Method Details

#[](key) ⇒ Object 



7
8
9
 
# File 'lib/informers/configs.rb', line 7

def [](key)
  @config_json[key.to_s]
end

#[]=(key, value) ⇒ Object 



11
12
13
 
# File 'lib/informers/configs.rb', line 11

def []=(key, value)
  @config_json[key.to_s] = value
end

#to_h ⇒ Object 



15
16
17
 
# File 'lib/informers/configs.rb', line 15

def to_h
  @config_json.to_h
end