ford442 commited on
Commit
e4f7018
·
verified ·
1 Parent(s): 173fa5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -96,7 +96,7 @@ refiner.scheduler=EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.
96
 
97
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
98
  tokenizer.tokenizer_legacy=False
99
- model = AutoModelForC5ausalLM.from_pretrained(checkpoint, device_map='balanced')
100
  #model = torch.compile(model)
101
 
102
  pipe.init_ipadapter(
 
96
 
97
  tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
98
  tokenizer.tokenizer_legacy=False
99
+ model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='balanced')
100
  #model = torch.compile(model)
101
 
102
  pipe.init_ipadapter(