Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	GPU最適化の適用条件を強化し、`torch.nn.Module`の場合のみ最適化を行うように修正しました。エラーハンドリングを追加し、最適化のスキップ理由をログに出力するようにしました。
Browse files- api_server.py +10 -2
- app_optimized.py +10 -3
    	
        api_server.py
    CHANGED
    
    | @@ -92,9 +92,17 @@ async def startup_event(): | |
| 92 | 
             
                    from stream_pipeline_offline import StreamSDK
         | 
| 93 | 
             
                    SDK = StreamSDK(cfg_pkl, data_root)
         | 
| 94 |  | 
| 95 | 
            -
                    # GPU | 
| 96 | 
             
                    if hasattr(SDK, 'decode_f3d') and hasattr(SDK.decode_f3d, 'decoder'):
         | 
| 97 | 
            -
                         | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 98 |  | 
| 99 | 
             
                    print("✅ SDK initialized with optimizations")
         | 
| 100 | 
             
                except Exception as e:
         | 
|  | |
| 92 | 
             
                    from stream_pipeline_offline import StreamSDK
         | 
| 93 | 
             
                    SDK = StreamSDK(cfg_pkl, data_root)
         | 
| 94 |  | 
| 95 | 
            +
                    # GPU最適化を適用(torch.nn.Moduleの場合のみ)
         | 
| 96 | 
             
                    if hasattr(SDK, 'decode_f3d') and hasattr(SDK.decode_f3d, 'decoder'):
         | 
| 97 | 
            +
                        try:
         | 
| 98 | 
            +
                            import torch.nn as nn
         | 
| 99 | 
            +
                            if isinstance(SDK.decode_f3d.decoder, nn.Module):
         | 
| 100 | 
            +
                                SDK.decode_f3d.decoder = gpu_optimizer.optimize_model(SDK.decode_f3d.decoder)
         | 
| 101 | 
            +
                                print("✅ Decoder model optimized")
         | 
| 102 | 
            +
                            else:
         | 
| 103 | 
            +
                                print("ℹ️ Decoder is not nn.Module, skipping optimization")
         | 
| 104 | 
            +
                        except Exception as e:
         | 
| 105 | 
            +
                            print(f"⚠️ Skipping GPU optimization: {e}")
         | 
| 106 |  | 
| 107 | 
             
                    print("✅ SDK initialized with optimizations")
         | 
| 108 | 
             
                except Exception as e:
         | 
    	
        app_optimized.py
    CHANGED
    
    | @@ -74,10 +74,17 @@ try: | |
| 74 | 
             
                SDK = StreamSDK(cfg_pkl, data_root)
         | 
| 75 | 
             
                print("✅ SDK初期化成功(最適化版)")
         | 
| 76 |  | 
| 77 | 
            -
                # GPU | 
| 78 | 
             
                if hasattr(SDK, 'decode_f3d') and hasattr(SDK.decode_f3d, 'decoder'):
         | 
| 79 | 
            -
                     | 
| 80 | 
            -
             | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 81 |  | 
| 82 | 
             
            except Exception as e:
         | 
| 83 | 
             
                print(f"❌ SDK初期化エラー: {e}")
         | 
|  | |
| 74 | 
             
                SDK = StreamSDK(cfg_pkl, data_root)
         | 
| 75 | 
             
                print("✅ SDK初期化成功(最適化版)")
         | 
| 76 |  | 
| 77 | 
            +
                # GPU最適化を適用(torch.nn.Moduleの場合のみ)
         | 
| 78 | 
             
                if hasattr(SDK, 'decode_f3d') and hasattr(SDK.decode_f3d, 'decoder'):
         | 
| 79 | 
            +
                    try:
         | 
| 80 | 
            +
                        import torch.nn as nn
         | 
| 81 | 
            +
                        if isinstance(SDK.decode_f3d.decoder, nn.Module):
         | 
| 82 | 
            +
                            SDK.decode_f3d.decoder = gpu_optimizer.optimize_model(SDK.decode_f3d.decoder)
         | 
| 83 | 
            +
                            print("✅ デコーダーモデルに最適化を適用")
         | 
| 84 | 
            +
                        else:
         | 
| 85 | 
            +
                            print("ℹ️ デコーダーはnn.Moduleではないため、最適化をスキップ")
         | 
| 86 | 
            +
                    except Exception as e:
         | 
| 87 | 
            +
                        print(f"⚠️ GPU最適化の適用をスキップ: {e}")
         | 
| 88 |  | 
| 89 | 
             
            except Exception as e:
         | 
| 90 | 
             
                print(f"❌ SDK初期化エラー: {e}")
         |