Update app.py

app.py

@@ -29,7 +29,7 @@ os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-nvcc')
 #subprocess.run(['pip', 'install', 'git+https://github.com/hidet-org/hidet.git'])
 #subprocess.run(['pip', 'install', 'git+https://github.com/ford442/hidet.git@thread'])
 
-os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
+#os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
 #os.system(f'{conda_bin}/conda install rcdr_py37::tensorrt')
 #subprocess.run(['sh', './hidet.sh'])
 import hidet
@@ -51,7 +51,7 @@ import cyper
 from image_gen_aux import UpscaleWithModel
 import torch
 import torch._dynamo
-torch._dynamo.list_backends()
+#torch._dynamo.list_backends()
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
@@ -372,7 +372,7 @@ def generate_30(
     #uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     with torch.no_grad():
-        torch.compiler.cudagraph_mark_step_begin()
+        #torch.compiler.cudagraph_mark_step_begin()
         rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_C_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
@@ -422,7 +422,7 @@ def generate_60(
     uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     with torch.no_grad():
-        torch.compiler.cudagraph_mark_step_begin()
+        #torch.compiler.cudagraph_mark_step_begin()
         rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_C_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
@@ -465,7 +465,7 @@ def generate_90(
     uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     with torch.no_grad():
-        torch.compiler.cudagraph_mark_step_begin()
+        #torch.compiler.cudagraph_mark_step_begin()
         rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_C_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)