Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 =
|
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(
|