update float based on cpu or gpu

app.py

@@ -46,14 +46,17 @@ IS_FLEXICUBES = os.path.basename(config_path).startswith('instant-mesh')
 # pick device
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 if device.type == 'cpu':
-    print("⚠️  No CUDA GPU detected. Falling back to CPU (very slow!)")
+    print("⚠️  No CUDA GPU detected. Falling back to CPU (this will be very slow!)")
+
+# choose torch dtype: float16 on GPU, float32 on CPU
+torch_dtype = torch.float16 if device.type == 'cuda' else torch.float32
 
 # ——— Load diffusion (Zero123) pipeline ———
 print("Loading diffusion model …")
 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2",
     custom_pipeline="zero123plus",
-    torch_dtype=torch.float16,
+    torch_dtype=torch_dtype,
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
     pipeline.scheduler.config, timestep_spacing='trailing'

requirements.txt

@@ -1,6 +1,6 @@
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
+torch
+torchvision
+torchaudio
 pytorch-lightning==2.1.2
 einops
 omegaconf