openseg

app.py

@@ -329,14 +329,14 @@ def construction_all():
         "WYBar/ART_test_weights",
         subfolder="fused_transformer", 
         torch_dtype=torch.bfloat16,
-        cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir"
+        # cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir"
     )
 
     pipeline = CustomFluxPipelineCfg.from_pretrained(
         "black-forest-labs/FLUX.1-dev",
         transformer=transformer,
         torch_dtype=torch.bfloat16,
-        cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir",
+        # cache_dir="/openseg_blob/v-yanbin/GradioDemo/cache_dir",
     ).to("cuda")
     # pipeline.enable_model_cpu_offload(gpu_id=0) # save vram