Merge pull request #129 from AgainstEntropy/main

Fix typo
cumulo-autumn · Apr 27, 2024 · 765d710 · 765d710
2 parents deabff6 + 87b5e04
commit 765d710
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/streamdiffusion/acceleration/tensorrt/__init__.py b/src/streamdiffusion/acceleration/tensorrt/__init__.py
@@ -169,13 +169,13 @@ def accelerate_with_tensorrt(
 
     del vae
 
-    cuda_steram = cuda.Stream()
+    cuda_stream = cuda.Stream()
 
-    stream.unet = UNet2DConditionModelEngine(unet_engine_path, cuda_steram, use_cuda_graph=use_cuda_graph)
+    stream.unet = UNet2DConditionModelEngine(unet_engine_path, cuda_stream, use_cuda_graph=use_cuda_graph)
     stream.vae = AutoencoderKLEngine(
         vae_encoder_engine_path,
         vae_decoder_engine_path,
-        cuda_steram,
+        cuda_stream,
         stream.pipe.vae_scale_factor,
         use_cuda_graph=use_cuda_graph,
     )

diff --git a/utils/wrapper.py b/utils/wrapper.py
@@ -599,18 +599,18 @@ def create_prefix(
                         else stream.frame_bff_size,
                     )
 
-                cuda_steram = cuda.Stream()
+                cuda_stream = cuda.Stream()
 
                 vae_config = stream.vae.config
                 vae_dtype = stream.vae.dtype
 
                 stream.unet = UNet2DConditionModelEngine(
-                    unet_path, cuda_steram, use_cuda_graph=False
+                    unet_path, cuda_stream, use_cuda_graph=False
                 )
                 stream.vae = AutoencoderKLEngine(
                     vae_encoder_path,
                     vae_decoder_path,
-                    cuda_steram,
+                    cuda_stream,
                     stream.pipe.vae_scale_factor,
                     use_cuda_graph=False,
                 )