Spaces:

GlyphByT5
/

ART_v1.0

Running on Zero

WYBar commited on 24 days ago

Commit

9a8a8da

1 Parent(s): 08acebb

fix /openseg

Files changed (1) hide show

modeling_crello.py CHANGED Viewed

@@ -153,7 +153,7 @@ class CrelloModel(PreTrainedModel):
         # device_map=device_map,
         trust_remote_code=True,
         torch_dtype=torch.bfloat16,
-        cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir",
       )
       # self.lm = AutoModelForCausalLM.from_pretrained(
       #   opt_version,

         # device_map=device_map,
         trust_remote_code=True,
         torch_dtype=torch.bfloat16,
+        # cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir",
       )
       # self.lm = AutoModelForCausalLM.from_pretrained(
       #   opt_version,