enable foreign code

models/image_generator.py

@@ -36,7 +36,8 @@ class OmniGenImageGenerator:
                     self.model_id,
                     torch_dtype=torch_dtype,
                     use_safetensors=True,
-                    variant="fp16" if self.device == "cuda" else None
+                    variant="fp16" if self.device == "cuda" else None,
+                    trust_remote_code=True
                 )
                 
                 # Apply optimizations
@@ -79,7 +80,8 @@ class OmniGenImageGenerator:
         self.pipeline = StableDiffusionPipeline.from_pretrained(
             self.model_id,
             torch_dtype=torch_dtype,
-            use_safetensors=True
+            use_safetensors=True,
+            trust_remote_code=True
         )
         
         if self.device == "cuda" and self.enable_cpu_offload:

models/model_3d_generator.py

@@ -58,7 +58,8 @@ class Hunyuan3DGenerator:
                     model_id,
                     torch_dtype=torch_dtype,
                     low_cpu_mem_usage=True,
-                    device_map="auto" if self.device == "cuda" else None
+                    device_map="auto" if self.device == "cuda" else None,
+                    trust_remote_code=True
                 )
                 
                 if self.device == "cpu":