Maximofn commited on
Commit
49f3202
·
1 Parent(s): 8849f7c

feat(src): :rocket: Debug device

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -29,6 +29,7 @@ def generate_video(
29
  device_map="balanced",
30
  torch_dtype=torch.float16,
31
  )
 
32
 
33
  # Cargar el pipeline
34
  pipeline = HunyuanVideoPipeline.from_pretrained(
@@ -37,6 +38,7 @@ def generate_video(
37
  torch_dtype=torch.float16,
38
  device_map="balanced",
39
  )
 
40
 
41
  # Generar el video usando el pipeline
42
  video = pipeline(
 
29
  device_map="balanced",
30
  torch_dtype=torch.float16,
31
  )
32
+ print(f"transformer_3bit device: {transformer_3bit.device}")
33
 
34
  # Cargar el pipeline
35
  pipeline = HunyuanVideoPipeline.from_pretrained(
 
38
  torch_dtype=torch.float16,
39
  device_map="balanced",
40
  )
41
+ print(f"pipeline device: {pipeline.device}")
42
 
43
  # Generar el video usando el pipeline
44
  video = pipeline(