Spaces:

wondervictor
/

ControlAR

Running on Zero

wondervictor commited on Nov 4, 2024

Commit

627dbc0

verified ·

1 Parent(s): 3094fff

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -88,7 +88,7 @@ class Model:
         precision = torch.bfloat16
         t5_model = T5Embedder(
             device="cuda",
-            local_cache=False,
             cache_dir='checkpoints/flan-t5-xl',
             dir_or_name='flan-t5-xl',
             torch_dtype=precision,

         precision = torch.bfloat16
         t5_model = T5Embedder(
             device="cuda",
+            local_cache=True,
             cache_dir='checkpoints/flan-t5-xl',
             dir_or_name='flan-t5-xl',
             torch_dtype=precision,