Update app.py

app.py

@@ -43,9 +43,9 @@ STYLE_PRESETS = {
     "Artistic": "artistic style, creative composition, unique visual style, expressive animation, stylized rendering"
 }
 
-# 固定模型配置 - 使用官方Diffusers兼容版本
-FIXED_MODEL = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"  # 最新版本
-# 备用选择: "Wan-AI/Wan2.1-T2V-14B-Diffusers"
+# 固定模型配置 - 使用NSFW版本的safetensors文件
+FIXED_MODEL = "https://huggingface.co/NSFW-API/NSFW_Wan_14b/resolve/main/wan_14B_e15.safetensors"
+BASE_CONFIG_MODEL = "Wan-AI/Wan2.1-T2V-14B"  # 用于获取配置信息
 
 # 质量增强提示词 - 适配视频
 QUALITY_ENHANCERS = [
@@ -115,31 +115,47 @@ def initialize_model():
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"🖥️ Using device: {device}")
         
-        print(f"📦 Loading Official Wan T2V model: {FIXED_MODEL}")
+        print(f"📦 Loading NSFW Wan T2V checkpoint: wan_14B_e15.safetensors")
+        print(f"🔧 Using config from: {BASE_CONFIG_MODEL}")
         
-        # 基础模型加载 - 使用官方Wan Pipeline
+        # 使用from_single_file方法加载NSFW检查点
         try:
             from diffusers import WanPipeline
-            pipeline = WanPipeline.from_pretrained(
+            pipeline = WanPipeline.from_single_file(
                 FIXED_MODEL,
+                config=BASE_CONFIG_MODEL,  # 使用官方配置
                 torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-                variant="fp16" if torch.cuda.is_available() else None,
                 use_safetensors=True,
                 safety_checker=None,
                 requires_safety_checker=False
             )
+            print("✅ Successfully loaded with WanPipeline.from_single_file()")
         except ImportError:
             # 如果WanPipeline不可用，使用通用DiffusionPipeline
-            print("⚠️ WanPipeline not found, using DiffusionPipeline")
-            pipeline = DiffusionPipeline.from_pretrained(
+            print("⚠️ WanPipeline not found, trying DiffusionPipeline")
+            pipeline = DiffusionPipeline.from_single_file(
                 FIXED_MODEL,
+                config=BASE_CONFIG_MODEL,  # 使用官方配置
+                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+                use_safetensors=True,
+                safety_checker=None,
+                requires_safety_checker=False,
+                trust_remote_code=True
+            )
+            print("✅ Successfully loaded with DiffusionPipeline.from_single_file()")
+        except Exception as single_file_error:
+            # 最后备选：尝试直接从官方模型仓库加载（但这不是NSFW版本）
+            print(f"⚠️ Single file loading failed: {single_file_error}")
+            print("🔄 Falling back to official model repository...")
+            pipeline = DiffusionPipeline.from_pretrained(
+                BASE_CONFIG_MODEL,
                 torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-                variant="fp16" if torch.cuda.is_available() else None,
                 use_safetensors=True,
                 safety_checker=None,
                 requires_safety_checker=False,
                 trust_remote_code=True
             )
+            print("✅ Loaded official model as fallback")
         
         pipeline = pipeline.to(device)
         
@@ -619,7 +635,7 @@ def create_interface():
             # 简化警告信息
             gr.HTML('''
                 <div class="warning-box">
-                    ⚠️ 18+ CONTENT WARNING ⚠️ | T2V Model: NSFW_Wan_14b
+                    ⚠️ 18+ CONTENT WARNING ⚠️ | T2V Model: NSFW_Wan_14b (wan_14B_e15.safetensors)
                 </div>
             ''')
             
@@ -847,8 +863,9 @@ def create_interface():
 # ===== 启动应用 =====
 if __name__ == "__main__":
     print("🎬 Starting NSFW Video Generator...")
-    print(f"🔧 Official Wan T2V Model: {FIXED_MODEL}")
-    print("🔧 Using official Wan-AI Diffusers-compatible model")
+    print(f"🔧 NSFW Wan T2V Checkpoint: wan_14B_e15.safetensors")
+    print(f"🔧 Base Config Model: {BASE_CONFIG_MODEL}")
+    print("🔧 Using from_single_file() method for NSFW checkpoint")
     print(f"🔧 Default Duration: {VIDEO_CONFIG['default_duration']}s")
     print(f"🔧 Default Resolution: {VIDEO_CONFIG['default_width']}×{VIDEO_CONFIG['default_height']}")
     print(f"🔧 Spaces GPU: {'✅ Available' if SPACES_AVAILABLE else '❌ Not Available'}")