qtnx commited on
Commit
a3a65d0
1 Parent(s): 7a3118c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,7 +23,7 @@ import torch
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
  from PIL import Image
25
 
26
- model_id = "qresearch/doubutsu-2b-pt-756"
27
  model = AutoModelForCausalLM.from_pretrained(
28
  model_id,
29
  trust_remote_code=True,
@@ -35,7 +35,7 @@ tokenizer = AutoTokenizer.from_pretrained(
35
  use_fast=True,
36
  )
37
 
38
- model.load_adapter("qresearch/doubutsu-2b-lora-756-docci")
39
 
40
  image = Image.open("IMAGE")
41
 
 
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
  from PIL import Image
25
 
26
+ model_id = "qresearch/doubutsu-2b-pt-378"
27
  model = AutoModelForCausalLM.from_pretrained(
28
  model_id,
29
  trust_remote_code=True,
 
35
  use_fast=True,
36
  )
37
 
38
+ model.load_adapter("qresearch/doubutsu-2b-lora-378-docci")
39
 
40
  image = Image.open("IMAGE")
41