Spaces:

wondervictor
/

ControlAR

Running on Zero

wondervictor commited on Nov 4, 2024

Commit

6e50755

verified ·

1 Parent(s): b67a03c

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -94,7 +94,7 @@ class Model:
         t5_model = T5Embedder(
             device=self.device,
             local_cache=True,
-            cache_dir='checkpoints/flan-t5-xl-1',
             dir_or_name='flan-t5-xl',
             torch_dtype=precision,
             model_max_length=120,

         t5_model = T5Embedder(
             device=self.device,
             local_cache=True,
+            cache_dir='checkpoints/flan-t5-xl',
             dir_or_name='flan-t5-xl',
             torch_dtype=precision,
             model_max_length=120,